mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
c708770b42
* Move `resize_pos_embed` to `mmcls.models.utils` * Refactor Vision Transformer * Refactor DeiT * Refactor MLP-Mixer * Refactor Swin-Transformer * Remove `indexing` arg * Support dynamic inputs for t2t_vit * Add copyright * Fix bugs in swin transformer * Add `pad_small_maps` option * Update swin transformer * Handle `attn_mask` in checkpoints of swin * Imporve by comments
18 lines
833 B
Python
18 lines
833 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .attention import MultiheadAttention, ShiftWindowMSA
|
|
from .augment.augments import Augments
|
|
from .channel_shuffle import channel_shuffle
|
|
from .embed import HybridEmbed, PatchEmbed, PatchMerging, resize_pos_embed
|
|
from .helpers import is_tracing, to_2tuple, to_3tuple, to_4tuple, to_ntuple
|
|
from .inverted_residual import InvertedResidual
|
|
from .make_divisible import make_divisible
|
|
from .position_encoding import ConditionalPositionEncoding
|
|
from .se_layer import SELayer
|
|
|
|
__all__ = [
|
|
'channel_shuffle', 'make_divisible', 'InvertedResidual', 'SELayer',
|
|
'to_ntuple', 'to_2tuple', 'to_3tuple', 'to_4tuple', 'PatchEmbed',
|
|
'PatchMerging', 'HybridEmbed', 'Augments', 'ShiftWindowMSA', 'is_tracing',
|
|
'MultiheadAttention', 'ConditionalPositionEncoding', 'resize_pos_embed'
|
|
]
|