############
ENOT modules
############

This page contains operation definitions which can be useful for search space
and regular model construction.

:red:`We do not recommend you to use this functional as it can be replaced by
autogeneration in most cases.`


**********
operations
**********

This module contains classes for searchable model construction. It contains
blocks from some popular architectures such as MobileNetV2 and ResNet.

conv_blocks
===========

.. autoclass:: enot.models.operations.conv_blocks.SearchableConv2d
    :members:
    :show-inheritance:

.. autoclass:: enot.models.operations.conv_blocks.SearchableFuseableSkipConv
    :members:
    :show-inheritance:

mobilenet_blocks
================

.. autoclass:: enot.models.operations.mobilenet_blocks.SearchableMobileInvertedBottleneck
    :members: __init__, forward
    :show-inheritance:

resnet_blocks
=============

.. autoclass:: enot.models.operations.resnet_blocks.SearchableResNetD
    :members:
    :show-inheritance:

.. autoclass:: enot.models.operations.resnet_blocks.SearchableResNetE
    :members:
    :show-inheritance:

resnext_blocks
==============

.. autoclass:: enot.models.operations.resnext_blocks.SearchableResNext
    :members:
    :show-inheritance:


*********
mobilenet
*********

This package contains classes for MobileNetV2-like model construction.

heads
=====

.. autoclass:: enot.models.mobilenet.heads.MobileNetBaseHead
    :members:
    :show-inheritance:

.. autoclass:: enot.models.mobilenet.heads.ArcfaceHead
    :members:
    :show-inheritance:

stems
=====

.. autoclass:: enot.models.mobilenet.stems.MobileNetBaseStem
    :members:
    :show-inheritance:


****
yolo
****

This package contains some classes for yolov5-like model construction.

yolo blocks
===========

.. autoclass:: enot.models.yolo.blocks.ConvBNActivation
    :members:
    :show-inheritance:

.. autoclass:: enot.models.yolo.blocks.Bottleneck
    :members:
    :show-inheritance:

.. autoclass:: enot.models.yolo.blocks.BottleNecksSequence
    :members:
    :show-inheritance:

.. autoclass:: enot.models.yolo.blocks.C3
    :members:
    :show-inheritance: