Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

This commit is contained in:
Adrian Wälchli 2021-12-16 18:02:50 +01:00 committed by GitHub
parent f37bd4677d
commit c335a7891d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch_lightning/plugins/training_type/tpu_spawn.py
View File

@ -125,9 +125,6 @@ class TPUSpawnPlugin(DDPSpawnPlugin):
if self.debug: if self.debug:
os.environ["PT_XLA_DEBUG"] = str(1) 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) shared_params = find_shared_parameters(self.model)
self.model_to_device() self.model_to_device()
if is_overridden("on_post_move_to_device", self.lightning_module): if is_overridden("on_post_move_to_device", self.lightning_module):