Minor log fix to avoid checkpoint and snapshot confusion (#940)

saumishr · facebook-github-bot · commit 47dd60373a8a · 2024-11-06T16:02:47.000-08:00
Summary: Pull Request resolved: #940 Minor log fix to avoid checkpoint and snapshot confusion Reviewed By: diego-urgell Differential Revision: D65554659 fbshipit-source-id: c61b934119561da933c777e324d0242a783dcda9
diff --git a/torchtnt/framework/callbacks/dcp_saver.py b/torchtnt/framework/callbacks/dcp_saver.py
@@ -361,7 +361,8 @@ def restore_with_id(
             )
 
         rank_zero_info(
-            f"Restored snapshot for checkpoint_id: {checkpoint_id}", logger=logger
+            f"Restored the checkpoint with checkpoint_id: {checkpoint_id}",
+            logger=logger,
         )
 
     @staticmethod

Original file line number	Diff line number	Diff line change
`@@ -361,7 +361,8 @@ def restore_with_id(`
`361`	`361`	`)`
`362`	`362`
`363`	`363`	`rank_zero_info(`
`364`		`- f"Restored snapshot for checkpoint_id: {checkpoint_id}", logger=logger`
	`364`	`+ f"Restored the checkpoint with checkpoint_id: {checkpoint_id}",`
	`365`	`+ logger=logger,`
`365`	`366`	`)`
`366`	`367`
`367`	`368`	`@staticmethod`