wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-25 12:50:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

added cpu example

Browse Source
This commit is contained in:
William Falcon committed 2019-07-08 17:45:09 -04:00
1 parent a311a62b48
commit c494e6d305
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/single_gpu_node_template.py
+1 -1
View File
@@ -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')