wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated docs

Browse Source
This commit is contained in:
William Falcon
2019-08-15 13:59:54 -04:00
parent 44da88fd15
commit 81837221a4
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+2 -1
View File
@@ -101,7 +101,8 @@ class CoolSystem(pl.LightningModule):
def configure_optimizers(self): def configure_optimizers(self):
# REQUIRED # REQUIRED
return [torch.optim.Adam(self.parameters(), lr=0.02)] # can return multiple optimizers and learning_rate schedulers
return torch.optim.Adam(self.parameters(), lr=0.02)
@pl.data_loader @pl.data_loader
def tng_dataloader(self): def tng_dataloader(self):