mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
readme
This commit is contained in:
@@ -77,7 +77,7 @@ class CoolModel(pl.LightningModule):
|
|||||||
|
|
||||||
def configure_optimizers(self):
|
def configure_optimizers(self):
|
||||||
# REQUIRED
|
# REQUIRED
|
||||||
return [torch.optim.Adam(self.parameters(), lr=0.02)]
|
return torch.optim.Adam(self.parameters(), lr=0.02)
|
||||||
|
|
||||||
@pl.data_loader
|
@pl.data_loader
|
||||||
def train_dataloader(self):
|
def train_dataloader(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user