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: