diff --git a/model_zoo/official/nlp/cpm/src/loss_monitor.py b/model_zoo/official/nlp/cpm/src/loss_monitor.py
index 5dc9a92e2b8..8136c9a6c72 100644
--- a/model_zoo/official/nlp/cpm/src/loss_monitor.py
+++ b/model_zoo/official/nlp/cpm/src/loss_monitor.py
@@ -16,6 +16,7 @@
 import time
 import math
 
+from mindspore import log as logger
 from mindspore.train.callback import Callback