This commit is contained in:
William Falcon
2019-10-05 17:18:26 -04:00
parent 3ac368dc62
commit 49c7d54dba
@@ -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):