From a48cccdc68cae5a4ba3fe465c936b812770c3ad8 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 27 Jul 2019 14:38:33 -0400 Subject: [PATCH] Update README.md --- README.md | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/README.md b/README.md index 40c2f0da..fadb2ec1 100644 --- a/README.md +++ b/README.md @@ -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):