diff --git a/tests/test_models.py b/tests/test_models.py index bdd31a5c..5ead6de8 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -66,6 +66,10 @@ def test_dp_output_reduce(): def test_model_saving_loading(): + """ + Tests use case where trainer saves the model, and user loads it from tags independently + :return: + """ hparams = get_hparams() model = LightningTestModel(hparams)