From 4c5802af61f8e838111f3beedf0755eacd972148 Mon Sep 17 00:00:00 2001 From: FlyingQianMM <245467267@qq.com> Date: Thu, 24 Sep 2020 04:01:03 +0000 Subject: [PATCH] add params name which cannnot been pruned of deeplabv3p_mobilenetv3_large --- paddlex/cv/models/slim/prune_config.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/paddlex/cv/models/slim/prune_config.py b/paddlex/cv/models/slim/prune_config.py index e9019f2..738c12f 100644 --- a/paddlex/cv/models/slim/prune_config.py +++ b/paddlex/cv/models/slim/prune_config.py @@ -304,7 +304,9 @@ def get_prune_params(model): 'conv9_depthwise_weights', 'conv6_expand_weights', 'conv5_se_2_weights', 'conv14_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(): if 'weight' not in param.name: -- GitLab