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')