mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
Update README.md
This commit is contained in:
1 parent
24255a9eab
commit
97f8291242
1 file changed
-1
@@ -56,7 +56,6 @@ class ExampleModel(RootModule):
|
||||
optimizer = self.choose_optimizer('adam', self.parameters(), {'lr': self.hparams.learning_rate}, 'optimizer')
|
||||
return [optimizer]
|
||||
|
||||
# LIGHTING WILL USE THE LOADERS YOU DEFINE HERE
|
||||
@property
|
||||
def tng_dataloader(self):
|
||||
return pytorch_dataloader('train')
|
||||
|
||||
Reference in new issue
Block a user