diff --git a/official/cv/resnet/train.py b/official/cv/resnet/train.py
index 3af39f99732c0d7a156ba5f3f8501ed9b88a20de..1f158548ca822c9366ec1acf6c3980f38bbdca02 100644
--- a/official/cv/resnet/train.py
+++ b/official/cv/resnet/train.py
@@ -357,7 +357,7 @@ def train_net():
 
     # define callbacks
     time_cb = TimeMonitor(data_size=step_size)
-    loss_cb = LossMonitor(config.has_trained_epoch)
+    loss_cb = LossCallBack(config.has_trained_epoch)
     cb = [time_cb, loss_cb]
     ckpt_save_dir = set_save_ckpt_dir()
     if config.save_checkpoint: