wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

testing slurm ddp

Browse Source
This commit is contained in:
William Falcon
2019-07-08 11:36:45 -04:00
parent 770aff5fc7
commit dbfcbbc004
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/trainer_gpu_cluster_template.py
+1 -1
View File
@@ -145,7 +145,7 @@ if __name__ == '__main__':
# use default args
root_dir = os.path.dirname(os.path.realpath(__file__))
log_dir = os.path.join(root_dir, 'pt_lightning_demo_logs')
checkpoint_dir = os.path.join(log_dir, 'model_weights')
checkpoint_dir = os.path.join(root_dir, 'model_weights')
parent_parser = HyperOptArgumentParser(strategy='grid_search', add_help=False)
# cluster args not defined inside the model