From f907158c97ca9eb182312b9a7ea7808bf77f3101 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 10:55:06 -0400 Subject: [PATCH] testing slurm ddp --- .../new_project_templates/trainer_gpu_cluster_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 2e029b7e..56fedd30 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -145,8 +145,8 @@ if __name__ == '__main__': parent_parser = get_default_parser(strategy='random_search', root_dir=root_dir) # cluster args not defined inside the model - parent_parser.add_argument('-gpu_partition', type=str) - parent_parser.add_argument('-per_experiment_nb_gpus', type=int) + parent_parser.add_argument('--gpu_partition', type=str) + parent_parser.add_argument('--per_experiment_nb_gpus', type=int) parent_parser.add_argument('--nb_gpu_nodes', type=int, default=1) # allow model to overwrite or extend args