updated docs

This commit is contained in:
William Falcon
2019-08-15 13:59:54 -04:00
parent 44da88fd15
commit 81837221a4
+2 -1
View File
@@ -101,7 +101,8 @@ class CoolSystem(pl.LightningModule):
def configure_optimizers(self):
# 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
def tng_dataloader(self):