This code requires version 0.4.0 of pytorch.

This code trains a VAE on BinaryMNIST based on https://github.com/pytorch/examples/tree/master/vae

To train a classic VAE with continuous latent variables:

python train_vae.py

To train a VAE with reinforce (20 latent variables with each 256 categories):

python train_reinforce_vae.py

To train a VAE with reinforce (20 latent variables with each 256 categories):

python train_gumbel_vae.py