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