mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
added slurm no process warning
This commit is contained in:
1 parent
c02b6c4c88
commit
4e67983f23
1 file changed
+3
@@ -449,6 +449,9 @@ 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
|
||||
|
||||
Reference in new issue
Block a user