/pytorch-mnist-VAE

Primary LanguageJupyter Notebook

pytorch-mnist-VAE

Variational AutoEncoder on the MNIST data set using the PyTorch

Dependencies

  • PyTorch
  • torchvision
  • numpy

Results

Generated samples from 2-D latent variable with random numbers from a normal distribution with mean 0 and variance 1

alt text

Reference

  1. Auto-Encoding Variational Bayes. Diederik P Kingma, Max Welling (paper): https://arxiv.org/abs/1312.6114
  2. 오토인코더의 모든 것 (slides): https://www.slideshare.net/NaverEngineering/ss-96581209
  3. Basic VAE Example (github): https://github.com/pytorch/examples/tree/master/vae
  4. hwalsuklee/tensorflow-mnist-VAE (github): https://github.com/hwalsuklee/tensorflow-mnist-VAE