Skip to content

This is a simple implementation of VAE on MNIST in pytorch

Notifications You must be signed in to change notification settings

r7rohan/Variational-Autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

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}
}

About

This is a simple implementation of VAE on MNIST in pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages