跟你的输入数据有关系
我在训练网络前又重新加载了一次数据
今天修改输入数据的时候,又出现了这个问题,经过上网查看,大家都说是数据索引超出了范围(出错最多的是标签label),我是训练数据使用填充时,把的值写大了(从0开始应该是2746,写成了2747),建议大家出现这个错误的时候把输入数据打印出来看看,是否超出了索引。
for epoch in range(5): # epoch设置小一点查看数据
total_loss = 0.0
for ii, data in enumerate(train_data_loader