瀏覽代碼

add params name which cannnot been pruned of deeplabv3p_mobilenetv3_large

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

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

@@ -304,7 +304,9 @@ def get_prune_params(model):
                 'conv9_depthwise_weights', 'conv6_expand_weights',
                 'conv9_depthwise_weights', 'conv6_expand_weights',
                 'conv5_se_2_weights', 'conv14_expand_weights',
                 'conv5_se_2_weights', 'conv14_expand_weights',
                 'conv4_depthwise_weights', 'conv7_expand_weights',
                 'conv4_depthwise_weights', 'conv7_expand_weights',
-                'conv7_depthwise_weights'
+                'conv7_depthwise_weights', 'encoder/aspp0/weights',
+                'decoder/merge/weights', 'encoder/image_pool/weights',
+                'decoder/weights'
             ]
             ]
         for param in program.global_block().all_parameters():
         for param in program.global_block().all_parameters():
             if 'weight' not in param.name:
             if 'weight' not in param.name: