diff --git a/docs/zh_cn/api_cn/dygraph/pruners/unstructured_pruner.rst b/docs/zh_cn/api_cn/dygraph/pruners/unstructured_pruner.rst index e990a54c75988f44934d42db0410c1a7337ae8b7..f532913fea08e91ade07d8d6411eec763a784326 100644 --- a/docs/zh_cn/api_cn/dygraph/pruners/unstructured_pruner.rst +++ b/docs/zh_cn/api_cn/dygraph/pruners/unstructured_pruner.rst @@ -45,6 +45,7 @@ UnstructuredPruner .. code-block:: python + import paddle from paddleslim import UnstructuredPruner from paddle.vision.models import LeNet as net import numpy as np @@ -67,6 +68,7 @@ UnstructuredPruner .. code-block:: python + import paddle from paddleslim import UnstructuredPruner from paddle.vision.models import LeNet as net import numpy as np @@ -101,6 +103,7 @@ UnstructuredPruner .. code-block:: python + import paddle from paddleslim import UnstructuredPruner from paddle.vision.models import LeNet as net import numpy as np @@ -129,6 +132,7 @@ UnstructuredPruner .. code-block:: python + import paddle from paddleslim import UnstructuredPruner from paddle.vision.models import LeNet as net import numpy as np diff --git a/docs/zh_cn/api_cn/static/prune/unstructured_prune_api.rst b/docs/zh_cn/api_cn/static/prune/unstructured_prune_api.rst index 1d2d080d577cac5a2b257dafb3980d9414887176..894b4240fd97c951c9269209d419183a539f5724 100644 --- a/docs/zh_cn/api_cn/static/prune/unstructured_prune_api.rst +++ b/docs/zh_cn/api_cn/static/prune/unstructured_prune_api.rst @@ -26,7 +26,7 @@ UnstrucuturedPruner .. code-block:: python - import paddle as paddle + import paddle import paddle.fluid as fluid from paddleslim.prune import UnstructuredPruner @@ -58,7 +58,7 @@ UnstrucuturedPruner .. code-block:: python - import paddle as paddle + import paddle import paddle.fluid as fluid from paddleslim.prune import UnstructuredPruner @@ -93,7 +93,7 @@ UnstrucuturedPruner .. code-block:: python - import paddle as paddle + import paddle import paddle.fluid as fluid from paddleslim.prune import UnstructuredPruner @@ -140,7 +140,7 @@ UnstrucuturedPruner .. code-block:: python - import paddle as paddle + import paddle import paddle.fluid as fluid from paddleslim.prune import UnstructuredPruner @@ -183,7 +183,7 @@ UnstrucuturedPruner .. code-block:: python - import paddle as paddle + import paddle import paddle.fluid as fluid from paddleslim.prune import UnstructuredPruner @@ -204,6 +204,7 @@ UnstrucuturedPruner exe = paddle.static.Executor(place) exe.run(startup_program) + pruner = UnstructuredPruner(paddle.static.default_main_program(), 'ratio', ratio=0.5, place=place) threshold = pruner.summarize_weights(paddle.static.default_main_program(), ratio=0.5) print(threshold)