show progressbar only on progress_rank 0 on ddp_slurm (#4437)

Co-authored-by: chaton <thomas@grid.ai> Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Lightning-AI · Nov 21, 2020 · 290998d · 290998d
1 parent 2582f96
commit 290998d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/accelerators/ddp_slurm_accelerator.py b/pytorch_lightning/accelerators/ddp_slurm_accelerator.py
@@ -123,7 +123,7 @@ def ddp_train(self, process_idx, model):
         self.set_world_ranks(process_idx)
 
         # toggle prog bar
-        if self.trainer.global_rank == 0 and self.trainer.progress_bar_callback is not None:
+        if self.trainer.global_rank != 0 and self.trainer.progress_bar_callback is not None:
             self.trainer.progress_bar_callback.disable()
 
         # set warning rank