瀏覽代碼

fix the ssld prune

sunyanfang01 5 年之前
父節點
當前提交
640b80f62c
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/models/slim/prune_config.py

+ 1 - 1
paddlex/cv/models/slim/prune_config.py

@@ -162,7 +162,7 @@ def get_prune_params(model):
                 continue
             prune_names.append(param.name)
     elif model_type.startswith("MobileNetV3"):
-        if model_type == 'MobileNetV3_small':
+        if model_type.startswith('MobileNetV3_small'):
             expand_prune_id = [3, 4]
         else:
             expand_prune_id = [2, 3, 4, 8, 9, 11]