提交 2a13b05f 编写于 作者: F FlyingQianMM

add params name which cannnot been pruned of deeplabv3p_mobilenetv3_large

上级 8a23df16
...@@ -304,7 +304,9 @@ def get_prune_params(model): ...@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册