diff --git a/paddleslim/analysis/__init__.py b/paddleslim/analysis/__init__.py index 76904c8d548208adb29188f28e9e0c6a0f11f30d..9caa0d24006a3e59f2d39c646d247b7e68480f96 100644 --- a/paddleslim/analysis/__init__.py +++ b/paddleslim/analysis/__init__.py @@ -15,9 +15,6 @@ import flops as flops_module from flops import * import model_size as model_size_module from model_size import * -import sensitive -from sensitive import * __all__ = [] __all__ += flops_module.__all__ __all__ += model_size_module.__all__ -__all__ += sensitive.__all__ diff --git a/paddleslim/prune/__init__.py b/paddleslim/prune/__init__.py index f8f87862f7c0e9c09c23b753be600eed5c915a90..b012254170d4d63bf24fcccaf8fa5f3eaeccac11 100644 --- a/paddleslim/prune/__init__.py +++ b/paddleslim/prune/__init__.py @@ -21,6 +21,8 @@ import controller_client from controller_client import * import sensitive_pruner from sensitive_pruner import * +import sensitive +from sensitive import * __all__ = [] __all__ += pruner.__all__ @@ -28,3 +30,4 @@ __all__ += auto_pruner.__all__ __all__ += controller_server.__all__ __all__ += controller_client.__all__ __all__ += sensitive_pruner.__all__ +__all__ += sensitive.__all__ diff --git a/paddleslim/analysis/sensitive.py b/paddleslim/prune/sensitive.py similarity index 100% rename from paddleslim/analysis/sensitive.py rename to paddleslim/prune/sensitive.py diff --git a/paddleslim/prune/sensitive_pruner.py b/paddleslim/prune/sensitive_pruner.py index 21f8900336d736de85fdaef42fb7479488dfe3ee..6213382fa9d47bae81c718f9f23c3e34146e05e4 100644 --- a/paddleslim/prune/sensitive_pruner.py +++ b/paddleslim/prune/sensitive_pruner.py @@ -19,7 +19,7 @@ from scipy.optimize import leastsq import numpy as np import paddle.fluid as fluid from ..common import get_logger -from ..analysis import sensitivity +from .sensitive import sensitivity from ..analysis import flops from .pruner import Pruner