Update engine.py (#767)

29fa4b2b · Jeff Rasley · GitHub · ee1ffe2e · 29fa4b2b
Unverified Commit 29fa4b2b authored Feb 18, 2021 by Jeff Rasley Committed by GitHub Feb 18, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

deepspeed/runtime/engine.py deepspeed/runtime/engine.py +3 -0

No files found.
--- a/deepspeed/runtime/engine.py
+++ b/deepspeed/runtime/engine.py
@@ -1482,6 +1482,9 @@ class DeepSpeedEngine(Module):
        if tag is None:
            tag = f"global_step{self.global_steps}"

+        # Ensure tag is a string
+        tag = str(tag)
+
        # Ensure checkpoint tag is consistent across ranks
        self._checkpoint_tag_validation(tag)