mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
fixed nccl init
This commit is contained in:
1 parent
5ba0a2ed48
commit
369174c4d3
1 file changed
+2
@@ -325,6 +325,8 @@ class Trainer(TrainerIO):
|
||||
node_rank = 0
|
||||
|
||||
# recover original exp before went into process
|
||||
# init in write mode only on proc 0
|
||||
self.experiment.debug = self.proc_rank > 0
|
||||
self.experiment = self.experiment.get_non_ddp_exp()
|
||||
|
||||
# show progbar only on prog_rank 0
|
||||
|
||||
Reference in new issue
Block a user