This commit is contained in:
William Falcon committed 2019-10-05 17:18:26 -04:00
1 parent 3ac368dc62
commit 49c7d54dba
1 file changed
+1 -1
@@ -77,7 +77,7 @@ class CoolModel(pl.LightningModule):
def configure_optimizers(self):
# REQUIRED
return [torch.optim.Adam(self.parameters(), lr=0.02)]
return torch.optim.Adam(self.parameters(), lr=0.02)
@pl.data_loader
def train_dataloader(self):