From f8ab6cc0ca97b0f8d987a0f1b1472be2dd280ecd Mon Sep 17 00:00:00 2001
From: FlyingQianMM <245467267@qq.com>
Date: Mon, 8 Jun 2020 19:21:15 +0800
Subject: [PATCH] fix unet cityscapes pretrain weights info

---
 paddlex/cv/models/utils/pretrain_weights.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlex/cv/models/utils/pretrain_weights.py b/paddlex/cv/models/utils/pretrain_weights.py
index 932092f..c8e0bb0 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']:
-- 
GitLab