wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-28 13:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated args

Browse Source
This commit is contained in:
William Falcon committed 2019-06-25 18:51:41 -04:00
1 parent de0f7fc936
commit d4ca295762
2 files changed
+7 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

docs/source/examples/fully_featured_trainer.py
+1
View File
@@ -98,6 +98,7 @@ def main(hparams, cluster, results_dict):
cluster=cluster,
checkpoint_callback=checkpoint,
early_stop_callback=early_stop,
gpus=hparams.gpus
)
# train model