【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