litingsjj opened this issue 4 years ago · 1 comments
Sorry, I have question about kfpn(F.softmax(outs, dim=-1)), the function of softmax is not used for channel?
F.softmax(outs, dim=-1)
forgot unsqueeze...