mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
Update lm_test_module.py
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user