【发布时间】:2019-10-16 20:58:58
【问题描述】:
我恢复了一个错误
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(), 1)
我不确定该怎么做,因为我尝试使用多个位置将其转换为 long。我试着放一个
.long
最后以及将 dtype 设置为 torch.long 仍然不起作用。
与此非常相似,但他没有做任何事情来得到答案 "Expected Long but got Int" while running PyTorch script
我已经更改了很多代码,这是我的最后一个版本,但现在给了我同样的问题。
def forward(self, inputs, targets):
"""
Args:
inputs: prediction matrix (before softmax) with shape (batch_size, num_classes)
targets: ground truth labels with shape (num_classes)
"""
log_probs = self.logsoftmax(inputs)
targets = torch.zeros(log_probs.size()).scatter_(1, targets.unsqueeze(1).data.cpu(), 1)
if self.use_gpu: targets = targets.to(torch.device('cuda'))
targets = (1 - self.epsilon) * targets + self.epsilon / self.num_classes
loss = (- targets * log_probs).mean(0).sum()
return loss
【问题讨论】:
-
您提到的问题大约是 5 个月大。这次他们不太可能只是坐在那里。对问题发表评论,看看他们是否找到了解决方案
-
python 3 不再支持 Long,所以请尝试使用 int(var_name)
标签: python python-3.x pytorch tensor