models.utils.rst

.. 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