gumbel
gumbel copied to clipboard
Correct loss
As seen here:
https://github.com/ericjang/gumbel-softmax/blob/master/Categorical%20VAE.ipynb
loss=tf.reduce_mean(-elbo)
Thanks, how did the outputs looks like in the notebook. Would you want to update it as well?