AS-GCN icon indicating copy to clipboard operation
AS-GCN copied to clipboard

Published 20 hours ago • verified publisher icon limaosen0

Hard gumbel softmax during training AIM?

Open tedhuang96 opened this issue 4 years ago • 0 comments

I am confused with this line of code. If hard is set as True even for training, how can the gradient flow back through AIM module in backprop?

https://github.com/limaosen0/AS-GCN/blob/ab6951c3e6fc1b22bb4a0dc87a33cca228aa69fe/net/utils/adj_learn.py#L264

tedhuang96 avatar May 17 '21 23:05 tedhuang96