Awesome Open Source
Awesome Open Source

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

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
jupyter-notebook (6,156
deep-learning (3,906
pytorch (2,318
generative-model (81
autoencoder (79
mnist (70
vae (69