提交 80ca5f9a 编写于 作者: C ceci3

fix py3

上级 7c40324b
...@@ -12,10 +12,7 @@ ...@@ -12,10 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from . import graph_wrapper from .graph_wrapper import GraphWrapper, VarWrapper, OpWrapper
from .graph_wrapper import * from .registry import Registry
from . import registry
from .registry import *
__all__ = graph_wrapper.__all__ __all__ = ['GraphWrapper', 'VarWrapper', 'OpWrapper', 'Registry']
__all__ += registry.__all__
...@@ -28,4 +28,4 @@ class SearchSpaceFactory(object): ...@@ -28,4 +28,4 @@ class SearchSpaceFactory(object):
""" """
assert isinstance(config_lists, list), "configs must be a list" assert isinstance(config_lists, list), "configs must be a list"
return CombineSearchSpace(config_lists) return CombineSearchSpace(config_lists)
...@@ -11,23 +11,14 @@ ...@@ -11,23 +11,14 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import pruner from .pruner import Pruner
from pruner import * from .auto_pruner import AutoPruner
import auto_pruner from .controller_server import ControllerServer
from auto_pruner import * from .controller_client import ControllerClient
import controller_server from .sensitive_pruner import SensitivePruner
from controller_server import * from .sensitive import sensitivity, flops_sensitivity
import controller_client
from controller_client import *
import sensitive_pruner
from sensitive_pruner import *
import sensitive
from sensitive import *
__all__ = [] __all__ = [
__all__ += pruner.__all__ 'Pruner', 'AutoPruner', 'ControllerServer', 'ControllerClient',
__all__ += auto_pruner.__all__ 'SensitivePruner', 'sensitivity', 'flops_sensitivity'
__all__ += controller_server.__all__ ]
__all__ += controller_client.__all__
__all__ += sensitive_pruner.__all__
__all__ += sensitive.__all__
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册