mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
cf2cb542f7
* Adjust vision transformer backbone architectures; * Add DropPath, trunc_normal_ for VisionTransformer implementation; * Add class token buring intermediate period and remove it during final period; * Fix some parameters loss bug; * * Store intermediate token features and impose no processes on them; * Remove class token and reshape entire token feature from NLC to NCHW; * Fix some doc error * Add a arg for VisionTransformer backbone to control if input class token into transformer; * Add stochastic depth decay rule for DropPath; * * Fix output bug when input_cls_token=False; * Add related unit test; * * Add arg: out_indices to control model output; * Add unit test for DropPath; * Apply suggestions from code review Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
14 lines
502 B
Python
14 lines
502 B
Python
from .drop import DropPath
|
|
from .inverted_residual import InvertedResidual, InvertedResidualV3
|
|
from .make_divisible import make_divisible
|
|
from .res_layer import ResLayer
|
|
from .se_layer import SELayer
|
|
from .self_attention_block import SelfAttentionBlock
|
|
from .up_conv_block import UpConvBlock
|
|
from .weight_init import trunc_normal_
|
|
|
|
__all__ = [
|
|
'ResLayer', 'SelfAttentionBlock', 'make_divisible', 'InvertedResidual',
|
|
'UpConvBlock', 'InvertedResidualV3', 'SELayer', 'DropPath', 'trunc_normal_'
|
|
]
|