Remove redundant special case for disabling the progress bar on TPU (#…

…11061)
Lightning-AI · Dec 16, 2021 · c335a78 · c335a78
1 parent f37bd46
commit c335a78
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/pytorch_lightning/plugins/training_type/tpu_spawn.py b/pytorch_lightning/plugins/training_type/tpu_spawn.py
@@ -125,9 +125,6 @@ def pre_dispatch(self, trainer: "pl.Trainer") -> None:
         if self.debug:
             os.environ["PT_XLA_DEBUG"] = str(1)
 
-        if self.tpu_global_core_rank != 0 and trainer.progress_bar_callback is not None:
-            trainer.progress_bar_callback.disable()
-
         shared_params = find_shared_parameters(self.model)
         self.model_to_device()
         if is_overridden("on_post_move_to_device", self.lightning_module):