ericjang / gumbel-softmax

categorical variational autoencoder using the Gumbel-Softmax estimator
MIT License
425 stars 101 forks source link

Discretization or not in the evaluation ? #5

Open Sunarker opened 6 years ago

Sunarker commented 6 years ago

Hi, Eric, I extent this code for evaluation. But I found the results in this paper can not be achieved if I discretized the output in the evaluation. Instead, if I keep the training and test consistent, the results are similar to those in the paper.