From 9f9d38673eb4a69e7357487bc2d4b73cf74ec288 Mon Sep 17 00:00:00 2001 From: williamFalcon Date: Fri, 6 Sep 2019 16:26:46 -0700 Subject: [PATCH] fixed demo --- examples/new_project_templates/multi_node_cluster_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/new_project_templates/multi_node_cluster_template.py b/examples/new_project_templates/multi_node_cluster_template.py index 9dffab06..f8be3f27 100644 --- a/examples/new_project_templates/multi_node_cluster_template.py +++ b/examples/new_project_templates/multi_node_cluster_template.py @@ -143,7 +143,7 @@ if __name__ == '__main__': # TODO: make 1 param parent_parser.add_argument('--per_experiment_nb_gpus', type=int, - help='how many gpus to use in a node') + default=2, help='how many gpus to use in a node') parent_parser.add_argument('--gpus', type=str, default='-1', help='how many gpus to use in the node')