updated docs

2019-08-15 13:59:54 -04:00 · 2019-08-15 13:59:54 -04:00 · 81837221a4
parent 44da88fd15
commit 81837221a4
1 changed files with 2 additions and 1 deletions
--- a/README.md
+++ b/README.md
@ -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):