0
点赞
收藏
分享

微信扫一扫

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


复现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 条评论