weiqiao/pytorch-mnist-vae

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

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