/Gaussian-Mixture-VAE

[Pytorch] Minimal implementation of a Variational Autoencoder (VAE) with Categorical Latent variables inspired from "Categorical Reparameterization with Gumbel-Softmax".

Primary LanguagePython

Watchers