Skip to content

Commit 5e09fd3

Browse files
cool425589tchatonawaelchli
authored
show progressbar only on progress_rank 0 on ddp_slurm (#4437)
Co-authored-by: chaton <[email protected]> Co-authored-by: Adrian Wälchli <[email protected]>
1 parent 9c8701f commit 5e09fd3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/accelerators/ddp_slurm_accelerator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def ddp_train(self, process_idx, model):
123123
self.set_world_ranks(process_idx)
124124

125125
# toggle prog bar
126-
if self.trainer.global_rank == 0 and self.trainer.progress_bar_callback is not None:
126+
if self.trainer.global_rank != 0 and self.trainer.progress_bar_callback is not None:
127127
self.trainer.progress_bar_callback.disable()
128128

129129
# set warning rank

0 commit comments

Comments
 (0)