安装的pytorch版本:
遇到的问题:
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)