51 lines
903 B
ReStructuredText
51 lines
903 B
ReStructuredText
|
.. role:: hidden
|
||
|
|
:class: hidden-section
|
||
|
|
|
||
|
|
mmcls.models.utils
|
||
|
|
===================================
|
||
|
|
|
||
|
|
This package includes some helper functions and common components used in various networks.
|
||
|
|
|
||
|
|
.. contents:: mmcls.models.utils
|
||
|
|
:depth: 2
|
||
|
|
:local:
|
||
|
|
:backlinks: top
|
||
|
|
|
||
|
|
.. currentmodule:: mmcls.models.utils
|
||
|
|
|
||
|
|
Common Components
|
||
|
|
------------------
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: generated
|
||
|
|
:nosignatures:
|
||
|
|
:template: classtemplate.rst
|
||
|
|
|
||
|
|
InvertedResidual
|
||
|
|
SELayer
|
||
|
|
ShiftWindowMSA
|
||
|
|
MultiheadAttention
|
||
|
|
ConditionalPositionEncoding
|
||
|
|
|
||
|
|
Helper Functions
|
||
|
|
------------------
|
||
|
|
|
||
|
|
channel_shuffle
|
||
|
|
^^^^^^^^^^^^^^^
|
||
|
|
.. autofunction:: channel_shuffle
|
||
|
|
|
||
|
|
make_divisible
|
||
|
|
^^^^^^^^^^^^^^
|
||
|
|
.. autofunction:: make_divisible
|
||
|
|
|
||
|
|
to_ntuple
|
||
|
|
^^^^^^^^^^^^^^
|
||
|
|
.. autofunction:: to_ntuple
|
||
|
|
.. autofunction:: to_2tuple
|
||
|
|
.. autofunction:: to_3tuple
|
||
|
|
.. autofunction:: to_4tuple
|
||
|
|
|
||
|
|
is_tracing
|
||
|
|
^^^^^^^^^^^^^^
|
||
|
|
.. autofunction:: is_tracing
|