Skip to content

Latest commit

 

History

History
22 lines (18 loc) · 849 Bytes

README.md

File metadata and controls

22 lines (18 loc) · 849 Bytes
Raw

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)

alt text

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