diff --git a/docs/zh_cn/api_cn/dygraph/pruners/l1norm_filter_pruner.rst b/docs/zh_cn/api_cn/dygraph/pruners/l1norm_filter_pruner.rst index d80097c507b6dd7abe142f69d905024e665ea2a5..2943ef16e921b68947fbe969bb76a38682420537 100644 --- a/docs/zh_cn/api_cn/dygraph/pruners/l1norm_filter_pruner.rst +++ b/docs/zh_cn/api_cn/dygraph/pruners/l1norm_filter_pruner.rst @@ -28,7 +28,7 @@ L1NormFilterPruner from paddleslim import L1NormFilterPruner net = mobilenet_v1(pretrained=False) pruner = L1NormFilterPruner(net, [1, 3, 224, 224]) - pruner.prune_var("conv2d_26.w_0", [0], pruned_ratio=0.5) + pruner.prune_var("conv2d_26.w_0", [0], 0.5) optimizer = paddle.optimizer.Momentum( learning_rate=0.1, parameters=net.parameters()) @@ -79,7 +79,7 @@ L1NormFilterPruner from paddleslim import L1NormFilterPruner net = mobilenet_v1(pretrained=False) pruner = L1NormFilterPruner(net, [1, 3, 224, 224]) - plan = pruner.prune_var("conv2d_26.w_0", [0]) + plan = pruner.prune_var("conv2d_26.w_0", [0], 0.5) print(f"plan: {plan}") paddle.summary(net, (1, 3, 224, 224)) ..