提交 05264ac2 编写于 作者: W wanghaoshuang

Merge branch 'develop' into 'develop'

Move sensitive from analysis package to prune package.

See merge request !43
...@@ -15,9 +15,6 @@ import flops as flops_module ...@@ -15,9 +15,6 @@ import flops as flops_module
from flops import * from flops import *
import model_size as model_size_module import model_size as model_size_module
from model_size import * from model_size import *
import sensitive
from sensitive import *
__all__ = [] __all__ = []
__all__ += flops_module.__all__ __all__ += flops_module.__all__
__all__ += model_size_module.__all__ __all__ += model_size_module.__all__
__all__ += sensitive.__all__
...@@ -21,6 +21,8 @@ import controller_client ...@@ -21,6 +21,8 @@ import controller_client
from controller_client import * from controller_client import *
import sensitive_pruner import sensitive_pruner
from sensitive_pruner import * from sensitive_pruner import *
import sensitive
from sensitive import *
__all__ = [] __all__ = []
__all__ += pruner.__all__ __all__ += pruner.__all__
...@@ -28,3 +30,4 @@ __all__ += auto_pruner.__all__ ...@@ -28,3 +30,4 @@ __all__ += auto_pruner.__all__
__all__ += controller_server.__all__ __all__ += controller_server.__all__
__all__ += controller_client.__all__ __all__ += controller_client.__all__
__all__ += sensitive_pruner.__all__ __all__ += sensitive_pruner.__all__
__all__ += sensitive.__all__
...@@ -19,7 +19,7 @@ from scipy.optimize import leastsq ...@@ -19,7 +19,7 @@ from scipy.optimize import leastsq
import numpy as np import numpy as np
import paddle.fluid as fluid import paddle.fluid as fluid
from ..common import get_logger from ..common import get_logger
from ..analysis import sensitivity from .sensitive import sensitivity
from ..analysis import flops from ..analysis import flops
from .pruner import Pruner from .pruner import Pruner
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册