torch.zeros(log_probs.size()).scatter_(1, targets.unsqueeze(1).data.cpu(), 1)报错

王远洋

关注

阅读 71

2023-02-23


复现HOReID代码时遇到的问题,详细报错:

Expected object of scalar type Long but got scalar type Int for argument #3 'index'

内外网都没有找到解决办法,我搞了一晚上,最后调通了(不知道后续有没有影响,至少当下是可以跑通代码了)

改成如下:

targets = torch.zeros(log_probs.size()).scatter_(1, (targets.unsqueeze(1).data.cpu()).long(), 1)

 

精彩评论(0)

0 0 举报