Skip to content

Commit

Permalink
show progressbar only on progress_rank 0 on ddp_slurm (#4437)
Browse files Browse the repository at this point in the history
Co-authored-by: chaton <thomas@grid.ai>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
  • Loading branch information
3 people authored Nov 6, 2020
1 parent 9c8701f commit 5e09fd3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_lightning/accelerators/ddp_slurm_accelerator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 5e09fd3

Please sign in to comment.