pytorch-mnist-CVAE

Conditional Variational AutoEncoder on the MNIST data set using the PyTroch

Dependencies

• PyTorch
• torchvision
• numpy

Results

Learned MNIST manifold with a condition of label (from 0 to 9)

Reference

1. Learning Structured Output Representation using Deep Conditional Generative Models (paper): https://papers.nips.cc/paper/5775-learning-structured-output-representation-using-deep-conditional-generative-modelsd
2. jojonki/AutoEncoders (github): https://github.com/jojonki/AutoEncoders/blob/master/cvae.ipynb
3. hujinsen/pytorch_VAE_CVAE (github): https://github.com/hujinsen/pytorch_VAE_CVAE/blob/master/CVAE.ipynb
4. Prasanna1991/pytorch-vae (github): https://github.com/Prasanna1991/pytorch-vae/blob/master/cvae.py