From fd7ce13ada0f7af77560c6e0d216f4192698d32b Mon Sep 17 00:00:00 2001 From: Jeff Rasley Date: Thu, 18 Feb 2021 14:07:45 -0800 Subject: [PATCH] Update engine.py --- deepspeed/runtime/engine.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/deepspeed/runtime/engine.py b/deepspeed/runtime/engine.py index c49b374bc054..b7d464252dc1 100755 --- a/deepspeed/runtime/engine.py +++ b/deepspeed/runtime/engine.py @@ -1482,6 +1482,9 @@ def save_checkpoint(self, save_dir, tag=None, client_state={}, save_latest=True) 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)