UserWarning: invalid index of a 0-dim tensor

将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告，它在提醒用户下个版本这将成为一个错误

1.

[python]  view plain  copy

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  
2.   train_loss += loss.data[0]  

修改方法如下：

[python]  view plain  copy

1. #原语句：  
2. train_loss+=loss.data[0]  
3. #修改后：  
4. train_loss+=loss.item()  
5. #bingo  

2.

[python]  view plain  copy

1. UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead.  
2.   label = Variable(label.cuda(), volatile=True)  

修改方法如下：

[python]  view plain  copy

1. #原语句  
2. label = Variable(label.cuda(), volatile=True)  
3. #修改后语句  
4.   label = Variable(label.cuda())  

转自：https://blog.youkuaiyun.com/weixin_41797117/article/details/80237179