SKAttention似乎有点小问题 #92
Description
单运行./model/attention/SKAttention.py时会报错，RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
目测应该是第40行U=sum(conv_outs)中调用sum时需要将tensor转numpy，而被转tensor参与梯度计算不能直接转？
{{ message }}