Contrib
================================================

.. automodule:: catalyst.contrib
    :members:
    :undoc-members:
    :show-inheritance:


Criterion
--------------------

.. automodule:: catalyst.contrib.criterion.ce
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.center
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.contrastive
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.dice
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.focal
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.huber
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.iou
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.lovasz
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.criterion.wing
    :members:
    :undoc-members:
    :show-inheritance:


Models
--------------------

Segmentation
~~~~~~~~~~~~~~~~

.. automodule:: catalyst.contrib.models.segmentation.unet
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.models.segmentation.linknet
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.models.segmentation.fpn
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.models.segmentation.psp
    :members:
    :undoc-members:
    :show-inheritance:


Modules
--------------------

.. automodule:: catalyst.contrib.modules.common
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.modules.lama
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.modules.noisy
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.modules.pooling
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.modules.real_nvp
    :members:
    :undoc-members:
    :show-inheritance:


Optimizers
--------------------

.. automodule:: catalyst.contrib.optimizers
    :members:
    :undoc-members:
    :show-inheritance:


Schedulers
--------------------

.. automodule:: catalyst.contrib.schedulers.base
    :members:
    :undoc-members:
    :show-inheritance:

.. automodule:: catalyst.contrib.schedulers.onecycle
    :members:
    :undoc-members:
    :show-inheritance: