Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update multi_node_own_slurm_script.py

This commit is contained in:
William Falcon 2019-09-09 10:54:43 -04:00 committed by GitHub
parent 3393086cb6
commit cbc619afa1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/new_project_templates/multi_node_examples/multi_node_own_slurm_script.py
View File

@ -41,7 +41,7 @@ def main(hparams):
# ------------------------ # ------------------------
trainer = Trainer( trainer = Trainer(
experiment=exp, experiment=exp,
gpus=[0, 1, 2, 3, 4, 5, 6, 7], gpus=8,
nb_gpu_nodes=2 nb_gpu_nodes=2
) )