2020-06-15 16:42:15 +08:00
|
|
|
from .channel_shuffle import channel_shuffle
|
2021-04-14 21:27:42 +08:00
|
|
|
from .cutmix import BatchCutMixLayer
|
2021-04-16 19:22:41 +08:00
|
|
|
from .helpers import to_2tuple, to_3tuple, to_4tuple, to_ntuple
|
2020-06-30 15:50:36 +08:00
|
|
|
from .inverted_residual import InvertedResidual
|
2020-06-15 16:42:15 +08:00
|
|
|
from .make_divisible import make_divisible
|
2021-02-25 14:06:58 +08:00
|
|
|
from .mixup import BatchMixupLayer
|
2020-06-30 15:50:36 +08:00
|
|
|
from .se_layer import SELayer
|
2020-06-15 16:42:15 +08:00
|
|
|
|
2021-02-25 14:06:58 +08:00
|
|
|
__all__ = [
|
|
|
|
|
'channel_shuffle', 'make_divisible', 'InvertedResidual', 'BatchMixupLayer',
|
2021-04-16 19:22:41 +08:00
|
|
|
'BatchCutMixLayer', 'SELayer', 'to_ntuple', 'to_2tuple', 'to_3tuple',
|
|
|
|
|
'to_4tuple'
|
2021-02-25 14:06:58 +08:00
|
|
|
]
|
