通过打印labels和predicted的数据发现,是因为batchsize分批不平均,导致predicted最后有一组只有16个数据。
而range()超过了16,所以报错。
解决办法:1.改变batchsize,使数据集分组刚刚好
2.填充最后一组不足的数据
通过打印labels和predicted的数据发现,是因为batchsize分批不平均,导致predicted最后有一组只有16个数据。
而range()超过了16,所以报错。
解决办法:1.改变batchsize,使数据集分组刚刚好
2.填充最后一组不足的数据