mmrazor/mmrazor/models/mutables/__init__.py

16 lines
749 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
from .derived_mutable import DerivedMutable
from .mutable_channel import (MutableChannel, OneShotMutableChannel,
SlimmableMutableChannel)
from .mutable_module import (DiffChoiceRoute, DiffMutableModule, DiffMutableOP,
OneHotMutableOP, OneShotMutableModule,
OneShotMutableOP)
from .mutable_value import MutableValue, OneShotMutableValue
__all__ = [
'OneShotMutableOP', 'OneShotMutableModule', 'DiffMutableOP',
'DiffChoiceRoute', 'DiffMutableModule', 'OneShotMutableChannel',
'SlimmableMutableChannel', 'MutableChannel', 'DerivedMutable',
'MutableValue', 'OneShotMutableValue', 'OneHotMutableOP'
]