Update lm_test_module.py

This commit is contained in:
William Falcon authored and GitHub committed 2019-07-26 14:39:04 -04:00
1 parent 98fcc17135
commit c7dab0d785
1 file changed
+2 -1
@@ -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