pytorch-lightning-generative-models

• Deep generative model training template using pytorch lightning and hydra.

models

• GAN
• DCGAN
• VAE

datasets

• MNIST
• CelebA

hydra

• hydra core (1.3.2) is applied.
• reference: https://github.com/ashleve/lightning-hydra-template
• Check out the easy and concise way provided by hyrda to instatiate needed objects in train.py