A Variational Auto Encoder on MNIST Dataset
Primary LanguageJupyter Notebook
Ported most of the code from https://github.com/sksq96/pytorch-vae Thanks buddy!