testing slurm ddp

2019-07-08 11:24:17 -04:00 · 2019-07-08 11:24:17 -04:00 · 1f3d011c58
parent a971ca515d
commit 1f3d011c58
1 changed files with 1 additions and 0 deletions
--- a/examples/new_project_templates/trainer_gpu_cluster_template.py
+++ b/examples/new_project_templates/trainer_gpu_cluster_template.py
@ -155,6 +155,7 @@ if __name__ == '__main__':
    parent_parser.add_argument('--test_tube_save_path', type=str, default=log_dir)
    parent_parser.add_argument('--experiment_name', type=str, default='pt_lightning_exp_a')
    parent_parser.add_argument('--model_save_path', type=str, default=checkpoint_dir)
+    parent_parser.add_argument('--gpus', type=str, default='-1')
    parent_parser.add_argument('--nb_hopt_trials', type=int, default=1)

    # allow model to overwrite or extend args