Skip to content

Commit 6f3a72f

Browse files
authored
Update pytorch_lightning/plugins/training_type/tpu_spawn.py
1 parent a2087c8 commit 6f3a72f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

pytorch_lightning/plugins/training_type/tpu_spawn.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -158,9 +158,7 @@ def save(self, state_dict: Dict, path: str) -> None:
158158
try:
159159
xm.save(state_dict, path)
160160
except RuntimeError as e:
161-
if "Failed to meet rendezvous" in str(e):
162-
pass
163-
else:
161+
if "Failed to meet rendezvous" not in str(e):
164162
raise e
165163

166164
def broadcast(self, obj: object, src: int = 0) -> object:

0 commit comments

Comments
 (0)