From c335a7891d82b647f1bebcce030e88a94ba52ac0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Thu, 16 Dec 2021 18:02:50 +0100 Subject: [PATCH] Remove redundant special case for disabling the progress bar on TPU (#11061) --- pytorch_lightning/plugins/training_type/tpu_spawn.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/pytorch_lightning/plugins/training_type/tpu_spawn.py b/pytorch_lightning/plugins/training_type/tpu_spawn.py index f6c85b060e..8b4f0536d6 100644 --- a/pytorch_lightning/plugins/training_type/tpu_spawn.py +++ b/pytorch_lightning/plugins/training_type/tpu_spawn.py @@ -125,9 +125,6 @@ class TPUSpawnPlugin(DDPSpawnPlugin): 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):