diff --git a/paddlex/cv/models/slim/prune_config.py b/paddlex/cv/models/slim/prune_config.py index e9019f21e874e3960fd19b133c89dc607ac1c147..738c12f4f1204578a0614d274c14c6ce3c901991 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: