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 slurm no process warning

Browse Source
This commit is contained in:
William Falcon
2019-07-18 17:06:56 -04:00
parent 4e67983f23
commit 5195124d4e
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
-3
View File
@@ -449,9 +449,6 @@ class Trainer(TrainerIO):
except KeyError as e:
node_rank = 0
print('-'*100)
print(gpu_nb)
print('-'*100)
# recover original exp before went into process
# init in write mode only on proc 0
self.experiment.debug = self.proc_rank > 0