torch遇到的一系列问题

1.报错原因: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()

2.原因分析:是具体的数字,不能加索引

解决方法:

原句:total_loss[0]/training_data.sents_size

改为:total_loss/training_data.sents_size

3.

错误分析:训练的批量样本数据输入值需要是long值的Tensor数据,而不是int值的Tensor数据.

解决:

后面加上long()

猜你喜欢

转载自blog.csdn.net/qq_41897073/article/details/117410403