Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update `dataloaders` params in example (#8191)

This commit is contained in:
Kaushik B 2021-06-29 17:53:48 +05:30 committed by GitHub
parent 6db0fe3659
commit f60aae9815
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pl_examples/bug_report_model.py
View File

@ -59,8 +59,8 @@ def run():
max_epochs=1,
weights_summary=None,
)
trainer.fit(model, train_dataloader=train_data, val_dataloaders=val_data)
trainer.test(model, test_dataloaders=test_data)
trainer.fit(model, train_dataloaders=train_data, val_dataloaders=val_data)
trainer.test(model, dataloaders=test_data)
if __name__ == '__main__':