pyraug.models.nn.rst

pyraug.models.nn

.. automodule::
   pyraug.models.nn

.. toctree::
   :maxdepth: 1

   pyraug.models.nn.BaseEncoder
   pyraug.models.nn.BaseDecoder
   pyraug.models.nn.BaseMetric

The Base Architectures

.. autosummary::
   ~pyraug.models.nn.BaseEncoder
   ~pyraug.models.nn.BaseDecoder
   ~pyraug.models.nn.BaseMetric
   :nosignatures:

Note

In case you want to provide your own neural architecture, make sure you make them inherit from these classes. See :ref:`making-your-own-vae`.