我正在使用model.fit_generator()
用Keras
训练神经网络。在拟合过程中,我已将steps_per_epoch
设置为16(len(training samples)/batch_size
)。
如果最小批次大小设置为12,并且训练样本总数为195,这是否意味着在训练阶段将不使用3个样本?
最佳答案
否,因为它是生成器,所以模型不知道训练样本的总数。因此,当到达由steps_per_epoch
参数定义的最后一步时,它将结束一个纪元。在您的情况下,实际上每个纪元将训练192个样本。
如果要使用模型中的所有样本,则可以在每个纪元开始时使用参数shuffle
来对数据进行混洗。