gumbel
gumbel copied to clipboard
Minimizing KL (not maximizing)
Thanks for uploading this. However, there is an error.
In the loss computation, you have elbo = data_dim * bce(x, x_hat) - KL
.
It should be elbo = data_dim * bce(x, x_hat) + KL
instead.
just noticed this too, can submit a pr if wanted