wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +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:22:10 -04:00
1 parent 35aa67df56
commit e12c8ad21a
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/fully_featured_trainer.py
+2 -1
View File
@@ -171,8 +171,9 @@ if __name__ == '__main__':
# format GPU layout
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# ---------------------
# RUN TRAINING
pdb.set_trace()
# ---------------------
# cluster and CPU
if hyperparams.on_cluster: