From c7dab0d7856f7035d5c4bdd0b37e550d49428599 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 14:39:04 -0400 Subject: [PATCH] Update lm_test_module.py --- pytorch_lightning/testing_models/lm_test_module.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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