3. torchgan.models

This models subpackage is a collection of popular GAN architectures. It has the support for existing architectures and provides a base class for extending to any form of new architecture. Currently the following models are supported:

You can construct a new model by simply calling its constructor.

>>> import torchgan.models as models
>>> dcgan_discriminator = DCGANDiscriminator()
>>> dcgan_generator = DCGANGenerator()

All models follow the same structure. There are additional customization options. Look into the individual documentation for such capabilities.

3.1. GAN

3.2. DCGAN

3.3. Conditional GAN

3.4. InfoGAN