把batch_sizes改小一些,或者看看是不是你的数据集有问题。
http://www.cocoachina.com/articles/93294
我也遇到同样的问题,你把数据集看清楚,图片和标签是否对应,你这个错误说明你的训练集或验证集或测试集里的图片不够5张,解决办法就是:1.你把数据集增加到大于5张或者2.你把batch_size的数量改为小于5等于5.
请问解决了么,我也碰到了这个问题,batchsize是1的时候可以正常跑,但是设大一点就总是报这个错,有时候第一个epoch正常跑完结果第二个epoch报错了,而且不管bs多大报的都是in batch 8