Variational-Autoencoder

Simple Variational autoencoder on MNIST

• Code in pytorch
• Latent space dimension can be changed
• Code for model, train, test

Reference

@misc{kingma2014autoencoding,
      title={Auto-Encoding Variational Bayes}, 
      author={Diederik P Kingma and Max Welling},
      year={2014},
      eprint={1312.6114},
      archivePrefix={arXiv},
      primaryClass={stat.ML}
}