Skip to content

Commit c335a78

Browse files
authored
Remove redundant special case for disabling the progress bar on TPU (#11061)
1 parent f37bd46 commit c335a78

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

pytorch_lightning/plugins/training_type/tpu_spawn.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,6 @@ def pre_dispatch(self, trainer: "pl.Trainer") -> None:
125125
if self.debug:
126126
os.environ["PT_XLA_DEBUG"] = str(1)
127127

128-
if self.tpu_global_core_rank != 0 and trainer.progress_bar_callback is not None:
129-
trainer.progress_bar_callback.disable()
130-
131128
shared_params = find_shared_parameters(self.model)
132129
self.model_to_device()
133130
if is_overridden("on_post_move_to_device", self.lightning_module):

0 commit comments

Comments
 (0)