From c494e6d305dc3595f6e005a0c3cd2ded0ae8b973 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 17:45:09 -0400 Subject: [PATCH] added cpu example --- examples/new_project_templates/single_gpu_node_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/new_project_templates/single_gpu_node_template.py b/examples/new_project_templates/single_gpu_node_template.py index a03be32f..34dc4441 100644 --- a/examples/new_project_templates/single_gpu_node_template.py +++ b/examples/new_project_templates/single_gpu_node_template.py @@ -95,7 +95,7 @@ if __name__ == '__main__': parent_parser = HyperOptArgumentParser(strategy='grid_search', add_help=False) # gpu args - parent_parser.add_argument('--gpus', type=str, default='-1', help='how many gpus to use in the node') + parent_parser.add_argument('--gpus', type=str, default='-1', help='how many gpus to use in the node. -1 uses all the gpus on the node') parent_parser.add_argument('--test_tube_save_path', type=str, default=test_tube_dir, help='where to save logs') 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')