diff --git a/pytorch_lightning/testing_models/lm_test_module.py b/pytorch_lightning/testing_models/lm_test_module.py index e33ee53e..158f8fbf 100644 --- a/pytorch_lightning/testing_models/lm_test_module.py +++ b/pytorch_lightning/testing_models/lm_test_module.py @@ -190,8 +190,9 @@ class LightningTestModel(LightningModule): return whatever optimizers we want here :return: list of optimizers """ + # try no scheduler for this model (testing purposes) optimizer = optim.Adam(self.parameters(), lr=self.hparams.learning_rate) - return [optimizer] + return [optimizer], [] def __dataloader(self, train): # init data generators