From b5ae41a08a17baf2195ffac85f9bc7ed1af2043a Mon Sep 17 00:00:00 2001 From: Saurabh Mishra Date: Wed, 6 Nov 2024 12:21:18 -0800 Subject: [PATCH] Minor log fix to avoid checkpoint and snapshot confusion Summary: Minor log fix to avoid checkpoint and snapshot confusion Reviewed By: diego-urgell Differential Revision: D65554659 --- torchtnt/framework/callbacks/dcp_saver.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torchtnt/framework/callbacks/dcp_saver.py b/torchtnt/framework/callbacks/dcp_saver.py index cdef922958..3c8fd3044a 100644 --- 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