diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 98cf87af..2f38cd0d 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -152,15 +152,18 @@ if __name__ == '__main__': parent_parser = HyperOptArgumentParser(strategy='grid_search', add_help=False) # 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('--nb_gpu_nodes', type=int, default=1) - parent_parser.add_argument('--test_tube_save_path', type=str, default=test_tube_dir) - parent_parser.add_argument('--slurm_log_path', type=str, default=slurm_out_dir) - parent_parser.add_argument('--model_save_path', type=str, default=checkpoint_dir) - parent_parser.add_argument('--experiment_name', type=str, default='pt_lightning_exp_a') - parent_parser.add_argument('--gpus', type=str, default='-1') - parent_parser.add_argument('--nb_hopt_trials', type=int, default=1) + parent_parser.add_argument('--gpu_partition', type=str, help='consult your cluster manual') + + # TODO: make 1 param + parent_parser.add_argument('--per_experiment_nb_gpus', type=int, 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') + + parent_parser.add_argument('--nb_gpu_nodes', type=int, default=1, help='how many nodes to use in a cluster') + parent_parser.add_argument('--test_tube_save_path', type=str, default=test_tube_dir, help='where to save logs') + parent_parser.add_argument('--slurm_log_path', type=str, default=slurm_out_dir, help='where to save slurm meta') + parent_parser.add_argument('--model_save_path', type=str, default=checkpoint_dir, help='where to save model') + parent_parser.add_argument('--experiment_name', type=str, default='pt_lightning_exp_a', help='test tube exp name') + parent_parser.add_argument('--nb_hopt_trials', type=int, default=1, help='how many grid search trials to run') # allow model to overwrite or extend args parser = LightningTemplateModel.add_model_specific_args(parent_parser, root_dir)