You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I think line 108 should be changed from return self.decode(z),F.softmax(q)
to return self.decode(z), F.softmax(q_y, dim=-1).view(-1, latent_dim * categorical_dim).
Since the softmax should be taken on categorical_dim sized vectors, and not on latent_dim * categorical_dim sized ones.
The text was updated successfully, but these errors were encountered:
Hi, thank you for sharing this project.
I think line 108 should be changed from
return self.decode(z),F.softmax(q)
to
return self.decode(z), F.softmax(q_y, dim=-1).view(-1, latent_dim * categorical_dim)
.Since the softmax should be taken on
categorical_dim
sized vectors, and not onlatent_dim * categorical_dim
sized ones.The text was updated successfully, but these errors were encountered: