MoritzWag / Representation-Learning

Representation Learning of Image Data with VAE.
2 stars 0 forks source link

CatVAE Implementation #21

Closed alexpiehler closed 4 years ago

alexpiehler commented 4 years ago

Implement a model class for categorical VAE with gumbel-softmax reparametrization trick based on the idea of https://arxiv.org/abs/1611.01144