Is log_softmax function missing in the Line 39 of generator.py?

[albertwy]

I think log_softmax function should be added in Line 39 if we want to use the Gumbel Softmax.

logits = F.log_softrmax(self.hidden(activ))

If I'm wrong, please let me know.
Thanks.