Warning volatile was removed and now has no effect

將pytorch更新到0.4.0最新版後對0.3.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  
2.   train_loss += loss.data[0]  
3. #原語句：  
4. train_loss+=loss.data[0]  
5. #修改後：  
6. train_loss+=loss.item()   

 

1. UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead.  
2. #原語句  
3. label = Variable(label.cuda(), volatile=True)  
4. #修改後語句  
5.   label = Variable(label.cuda())