Update pytorch_lightning/plugins/training_type/tpu_spawn.py

Lightning-AI · Mar 2, 2021 · 6f3a72f · 6f3a72f
1 parent a2087c8
commit 6f3a72f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pytorch_lightning/plugins/training_type/tpu_spawn.py b/pytorch_lightning/plugins/training_type/tpu_spawn.py
@@ -158,9 +158,7 @@ def save(self, state_dict: Dict, path: str) -> None:
  try:
  xm.save(state_dict, path)
  except RuntimeError as e:
- if "Failed to meet rendezvous" in str(e):
- pass
- else:
+ if "Failed to meet rendezvous" not in str(e):
  raise e
 
  def broadcast(self, obj: object, src: int = 0) -> object: