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

Update README.md

Browse Source
This commit is contained in:
William Falcon
2019-07-27 14:38:33 -04:00
committed by GitHub
parent 66188209b5
commit a48cccdc68
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1 -3
View File
@@ -77,9 +77,7 @@ class CoolModel(ptl.LightningModule):
return {'avg_val_loss': avg_loss}
def configure_optimizers(self):
optim = torch.optim.Adam(self.parameters(), lr=0.02)
self.optimizers = [optim]
return self.optimizers
return [torch.optim.Adam(self.parameters(), lr=0.02)]
@ptl.data_loader
def tng_dataloader(self):