瀏覽代碼

Merge pull request #1008 from HeQinWill/patch-1

fix the backbone message
will-jl944 4 年之前
父節點
當前提交
efc591fc78
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/models/detector.py

+ 1 - 1
paddlex/cv/models/detector.py

@@ -1329,7 +1329,7 @@ class PPYOLOv2(YOLOv3):
         if backbone not in ['ResNet50_vd_dcn', 'ResNet101_vd_dcn']:
             raise ValueError(
                 "backbone: {} is not supported. Please choose one of "
-                "('ResNet50_vd_dcn', 'ResNet18_vd')".format(backbone))
+                "('ResNet50_vd_dcn', 'ResNet101_vd_dcn')".format(backbone))
         self.backbone_name = backbone
 
         if paddlex.env_info['place'] == 'gpu' and paddlex.env_info[