Skip to content

Instantly share code, notes, and snippets.

@williamFalcon
Created March 18, 2020 15:38
Show Gist options
  • Save williamFalcon/666ce4a5210fac5cb024d0962e450a9e to your computer and use it in GitHub Desktop.
Save williamFalcon/666ce4a5210fac5cb024d0962e450a9e to your computer and use it in GitHub Desktop.
train = DataLoader(...)
test = DataLoader(...)
val = DataLoader(...)
trainer = Trainer()
model = LightningModule()
trainer.fit(model,
train_dataloader=train,
val_dataloader=val,
test_dataloader=test)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment