1. 程式人生 > 其它 >invalid index of a 0-dim tensor, Use tensor.item() to convert a 0-dim tensor to a number 報錯

invalid index of a 0-dim tensor, Use tensor.item() to convert a 0-dim tensor to a number 報錯

技術標籤:程式碼python

轉載:https://blog.csdn.net/weixin_41797117/article/details/80237179
將pytorch更新到0.4.0最新版後對0.3.1版本程式碼會有如下警告,它在提醒使用者下個版本這將成為一個錯誤
修改方法如下:

#原語句:
train_loss+=loss.data[0]
#修改後:
train_loss+=loss.item()
#bingo