tensorflow - 如果样本大小不能被 Keras 模型中的 batch_size 整除怎么办

Question

如果我们将批量大小指定为 15，并且在 Keras 模型训练中不能被 15 整除的样本大小为 1000，该怎么办？它还能训练吗？

我也看过这个答案，但这对问题没有帮助

请任何人都可以解释一下谢谢。

标签： tensorflowmachine-learningkeras

Answer 1

我找到了答案。如果是这种情况，它将把剩余的 10 个样本带到 epoch 的最后一步。

例如：15x66+10=1000这意味着它将需要 66 个大小为 15 的批次，而对于最后的步骤，它只需要 10 个。

无论如何，这仅适用于input_shape，如果我们使用batch_input_shape它将给我们一个错误，因为我们在图形级别指定批处理形状。