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