/Gumbel-Softmax-Implementation-with-VAE

The implementation of Gumbel softmax reparametrization trick for discrete VAE

Primary LanguagePython

Watchers