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

added single node example

Browse Source
This commit is contained in:
William Falcon
2019-07-08 17:31:47 -04:00
parent bd43c4417f
commit c750015c80
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+5 -1
View File
@@ -289,7 +289,11 @@ class Trainer(TrainerIO):
:return:
"""
# node rank using relative slurm id
node_rank = int(os.environ['SLURM_NODEID'])
# otherwise default to node rank 0
try:
node_rank = int(os.environ['SLURM_NODEID'])
except KeyError as e:
node_rank = 0
# recover original exp before went into process
self.experiment = self.experiment.get_non_ddp_exp()