我正在使用model.fit_generator()Keras训练神经网络。在拟合过程中,我已将steps_per_epoch设置为16(len(training samples)/batch_size)。

如果最小批次大小设置为12,并且训练样本总数为195,这是否意味着在训练阶段将不使用3个样本?

最佳答案

否,因为它是生成器,所以模型不知道训练样本的总数。因此,当到达由steps_per_epoch参数定义的最后一步时,它将结束一个纪元。在您的情况下,实际上每个纪元将训练192个样本。

如果要使用模型中的所有样本,则可以在每个纪元开始时使用参数shuffle来对数据进行混洗。

10-04 22:21
查看更多