复现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)