Browse Source

fix the backbone message

modify the backbone prompt message of the PPYOLOv2 model
何秦 4 years ago
parent
commit
8d4e3214dc
1 changed files with 1 additions and 1 deletions
  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[