【pytorch踩坑】IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor

報錯原因分析:

train_loss += loss.data[0] 是pytorch0.3.1版本代碼,在0.4-0.5版本的pytorch會出現警告,不會報錯,但是0.5版本以上的pytorch就會報錯,總的來說是版本更新問題.

解決方法:

#將原語句：

train_loss+=loss.data[0]

#修改爲：

train_loss+=loss.item()

#bingo
————————————————
版權聲明：本文爲CSDN博主「chen645096127」的原創文章，遵循CC 4.0 BY-SA版權協議，轉載請附上原文出處鏈接及本聲明。
原文鏈接：https://blog.csdn.net/chen645096127/article/details/94019443