- Notifications
You must be signed in to change notification settings - Fork107
timbmg/VAE-CVAE-MNIST
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
VAE paper:Auto-Encoding Variational Bayes
CVAE paper:Semi-supervised Learning with Deep Generative Models
In order to runconditional variational autoencoder, add--conditional to the the command. Check out the other commandline options in the code for hyperparameter settings (like learning rate, batch size, encoder/decoder layer depth and size).
All plots obtained after 10 epochs of training. Hyperparameters accordning to default settings in the code; not tuned.
The modeled latent distribution after 10 epochs and 100 samples per digit.
| VAE | CVAE |
|---|---|
 |  |
Randomly sampled z, and their output. For CVAE, each c has been given as input once.
| VAE | CVAE |
|---|---|
 |  |
