Variational Autoencoder using MNIST dataset
Colab Demo
Examples
Description
TODO
Links
Exploring Latent Space |
Some examples from test set |
Another one Variational Autoencoder trained on MNIST
Weights:
- Create VAE training pipeline and train
- Architecture
- BCE loss instead of MSE
- Convolutional Pooling instead of MaxPool
- Train Autoencoder
- Train Fully-Coonvolutional AE
- Train Variational Autoencoder
- Train Fully-Coonvolutional VAE
- Compare different architectures (Need some metric)
- Minimum working example in Colab
- Train VAE on different dataset (ex. CelebA)
Github: Pytorch Examples
Arxiv: Auto-Encoding Variational Bayes