mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
d510b8b174
* add layer scale * add layer scale * add layer scale * Update mmcv/cnn/bricks/transformer.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmcv/cnn/bricks/transformer.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add layer scale * move LayerScale * add layer_scale_init_value * add typehint * fix for tensor with any dim * fix layer scale rule * fix layer scale rule * fix test * add docs Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
71 lines
1.0 KiB
ReStructuredText
71 lines
1.0 KiB
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
mmcv.cnn
|
|
===================================
|
|
|
|
.. contents:: mmcv.cnn
|
|
:depth: 2
|
|
:local:
|
|
:backlinks: top
|
|
|
|
.. currentmodule:: mmcv.cnn
|
|
|
|
Module
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
ContextBlock
|
|
Conv2d
|
|
Conv3d
|
|
ConvAWS2d
|
|
ConvModule
|
|
ConvTranspose2d
|
|
ConvTranspose3d
|
|
ConvWS2d
|
|
DepthwiseSeparableConvModule
|
|
GeneralizedAttention
|
|
HSigmoid
|
|
HSwish
|
|
LayerScale
|
|
Linear
|
|
MaxPool2d
|
|
MaxPool3d
|
|
NonLocal1d
|
|
NonLocal2d
|
|
NonLocal3d
|
|
Scale
|
|
Swish
|
|
|
|
Build Function
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
build_activation_layer
|
|
build_conv_layer
|
|
build_norm_layer
|
|
build_padding_layer
|
|
build_plugin_layer
|
|
build_upsample_layer
|
|
|
|
Miscellaneous
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
fuse_conv_bn
|
|
conv_ws_2d
|
|
is_norm
|
|
make_res_layer
|
|
make_vgg_layer
|
|
get_model_complexity_info
|