diff --git a/paddlex/cv/models/utils/pretrain_weights.py b/paddlex/cv/models/utils/pretrain_weights.py
index 932092fa6423b4457b5487fdf69b5eca5bff2839..c8e0bb0255d19d991cf9cbf1707b1cd71dde5280 100644
--- a/paddlex/cv/models/utils/pretrain_weights.py
+++ b/paddlex/cv/models/utils/pretrain_weights.py
@@ -140,7 +140,7 @@ def get_pretrain_weights(flag, class_name, backbone, save_dir):
     elif flag == 'CITYSCAPES':
         model_name = '{}_{}'.format(class_name, backbone)
         if class_name == 'UNet':
-            logging.warning(warning_info.format(class_name, flag, 'IMAGENET'))
+            logging.warning(warning_info.format(class_name, flag, 'COCO'))
             flag = 'COCO'
         if class_name == 'HRNet' and backbone.split('_')[
                 -1] in ['W30', 'W32', 'W40', 'W48', 'W60', 'W64']: