报错原因分析:
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
报错原因分析:
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
上一篇:计算机视觉中的细节问题(九)
下一篇:什么是棋盘效应
我在是在多线程中出现了这
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M