From 1f3d011c58d975b54fbc40e4b7290ed3f02d8b69 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 11:24:17 -0400 Subject: [PATCH] testing slurm ddp --- examples/new_project_templates/trainer_gpu_cluster_template.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 356e8624b3..99d8834c69 100644 --- 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