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

Update multi_node_cluster_auto_slurm.py

Browse Source
This commit is contained in:
William Falcon
2019-10-04 13:05:52 -04:00
committed by GitHub
parent 42764d18c7
commit 5fdfad5766
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py
+4 -1
View File
@@ -82,7 +82,8 @@ def main(hparams, cluster):
checkpoint_callback=checkpoint,
early_stop_callback=early_stop,
gpus=hparams.per_experiment_nb_gpus,
nb_gpu_nodes=hyperparams.nb_gpu_nodes
nb_gpu_nodes=hyperparams.nb_gpu_nodes,
distributed_backend=hyperparams.distributed_backend
)
# ------------------------
@@ -168,6 +169,8 @@ if __name__ == '__main__':
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('--distributed_backend', type=str, default='ddp',
help='ddp or ddp2')
parent_parser.add_argument('--experiment_name', type=str, default='pt_lightning_exp_a',
help='test tube exp name')
parent_parser.add_argument('--num_hyperparam_trials', type=int, default=6,