A variational autoencoder with discrete latent variables, and inference using the concrete distributions.
Follows the paper The Concrete Distribution: A Continuous Relaxation of Discrete Random Variables.
tba
torch>=0.4.0
torchvision>=0.4.0
daandouwe/concrete-vae
A VAE with discrete latent variables and inference with the Concrete distribution.
Python