Models
===================================

U-net
~~~~~~~~~~~~~~~~~~~~

.. automodule:: synthtorch.models.unet
   :members:

Variational Autoencoder
~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: synthtorch.models.vae
   :members:

N-layer CNN
~~~~~~~~~~~~~~~~~~~~

.. automodule:: synthtorch.models.nconvnet
   :members:

SegAE
~~~~~~~~~~~~~~~~~~~~

.. automodule:: synthtorch.models.segae
   :members:

DenseNet
~~~~~~~~~~~~~~~~~~~~

.. automodule:: synthtorch.models.densenet
   :members: