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>
2020-11-06 08:36:22 +08:00 · 2020-11-06 08:36:22 +08:00 · 5e09fd31e9
parent 9c8701f2e2
commit 5e09fd31e9
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/accelerators/ddp_slurm_accelerator.py
+++ b/pytorch_lightning/accelerators/ddp_slurm_accelerator.py
@ -123,7 +123,7 @@ class DDPSLURMAccelerator(Accelerator):
        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