瀏覽代碼

remove lr_mult_list in mobilenet since ppcls no longer supports

will-jl944 4 年之前
父節點
當前提交
de3861b63d
共有 1 個文件被更改,包括 1 次插入3 次删除
  1. 1 3
      dygraph/paddlex/cv/models/classifier.py

+ 1 - 3
dygraph/paddlex/cv/models/classifier.py

@@ -577,9 +577,7 @@ class MobileNetV3_small(BaseClassifier):
         model_name = 'MobileNetV3_small_x' + str(float(scale)).replace('.',
         model_name = 'MobileNetV3_small_x' + str(float(scale)).replace('.',
                                                                        '_')
                                                                        '_')
         super(MobileNetV3_small, self).__init__(
         super(MobileNetV3_small, self).__init__(
-            model_name=model_name,
-            num_classes=num_classes,
-            lr_mult_list=[.1, .1, .2, .2, .3])
+            model_name=model_name, num_classes=num_classes)
 
 
 
 
 class MobileNetV3_small_ssld(BaseClassifier):
 class MobileNetV3_small_ssld(BaseClassifier):