Skip to content

GitLab

Unverified Commit 29fa4b2b authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

Update engine.py (#767)

parent ee1ffe2e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
deepspeed/runtime/engine.py
View file @ 29fa4b2b
...@@ -1482,6 +1482,9 @@ class DeepSpeedEngine(Module): ...@@ -1482,6 +1482,9 @@ class DeepSpeedEngine(Module):
if tag is None: if tag is None:
tag = f"global_step{self.global_steps}" tag = f"global_step{self.global_steps}"
# Ensure tag is a string
tag = str(tag)
# Ensure checkpoint tag is consistent across ranks # Ensure checkpoint tag is consistent across ranks
self._checkpoint_tag_validation(tag) self._checkpoint_tag_validation(tag)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment