From dce1baefb3a15ccdebd0cbb1eb2d0cec229d89be Mon Sep 17 00:00:00 2001
From: liujing04 <129054828+liujing04@users.noreply.github.com>
Date: Sun, 9 Apr 2023 15:52:48 +0000
Subject: [PATCH] Update utils.py

---
 train/utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train/utils.py b/train/utils.py
index de6ceef..71528d6 100644
--- a/train/utils.py
+++ b/train/utils.py
@@ -50,7 +50,7 @@ def load_checkpoint_d(checkpoint_path, combd,sbd, optimizer=None,load_opt=1):
       optimizer.load_state_dict(checkpoint_dict['optimizer'])
   #   except:
   #     traceback.print_exc()
-  logger.info("Loaded checkpoint '{}' (iteration {})" .format(checkpoint_path, iteration))
+  logger.info("Loaded checkpoint '{}' (epoch {})" .format(checkpoint_path, iteration))
   return model, optimizer, learning_rate, iteration
 
 
@@ -80,7 +80,7 @@ def load_checkpoint_d(checkpoint_path, combd,sbd, optimizer=None,load_opt=1):
 #     model.module.load_state_dict(new_state_dict)
 #   else:
 #     model.load_state_dict(new_state_dict)
-#   logger.info("Loaded checkpoint '{}' (iteration {})" .format(
+#   logger.info("Loaded checkpoint '{}' (epoch {})" .format(
 #     checkpoint_path, iteration))
 #   return model, optimizer, learning_rate, iteration
 def load_checkpoint(checkpoint_path, model, optimizer=None,load_opt=1):
@@ -116,7 +116,7 @@ def load_checkpoint(checkpoint_path, model, optimizer=None,load_opt=1):
       optimizer.load_state_dict(checkpoint_dict['optimizer'])
   #   except:
   #     traceback.print_exc()
-  logger.info("Loaded checkpoint '{}' (iteration {})" .format(checkpoint_path, iteration))
+  logger.info("Loaded checkpoint '{}' (epoch {})" .format(checkpoint_path, iteration))
   return model, optimizer, learning_rate, iteration