mistake in Gumbel softmax sample function.
Closed this issue · 0 comments
sobalgi commented
The gumbel_softmax_sample function is logits + gumbel_sample. But it should have been F.log_softmax(logits) + gumbel_sample according to the paper. Is this not a mistake?