From bba51dde8cd742aa243c33e33ac99a63ffbf5b34 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 14:27:19 -0400 Subject: [PATCH] updated parser help --- .../trainer_gpu_cluster_template.py | 21 +++++++++++-------- 1 file changed, 12 insertions(+), 9 deletions(-) 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)