pytorch警告解决方案

安装的pytorch版本:
pytorch警告解决方案_第1张图片

遇到的问题:
1.UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number losses.update(loss.data[0], inputs.size(0))
修改后:

    losses.update(loss.item(), inputs.size(0))

2.UserWarning: volatile was removed and now has no effect. Use with torch.no_grad(): instead.
inputs = Variable(inputs, volatile=True)
修改后:

    inputs = Variable(inputs)

你可能感兴趣的:(pytorch)