Преглед изворни кода

fix unet cityscapes pretrain weights info

FlyingQianMM пре 5 година
родитељ
комит
f8ab6cc0ca
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      paddlex/cv/models/utils/pretrain_weights.py

+ 1 - 1
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']: