其他分享
首页 > 其他分享> > IndexError: index 16 is out of bounds for dimension 0 with size 16

IndexError: index 16 is out of bounds for dimension 0 with size 16

作者:互联网

 通过打印labels和predicted的数据发现,是因为batchsize分批不平均,导致predicted最后有一组只有16个数据。

而range()超过了16,所以报错。

解决办法:1.改变batchsize,使数据集分组刚刚好

                  2.填充最后一组不足的数据 

标签:index,IndexError,一组,16,batchsize,报错,predicted,数据
来源: https://blog.csdn.net/liu121212123/article/details/120869561