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

Question

如果我们在 Keras 模型训练中指定批量大小为 15，样本大小为 1000，且不能被 15 整除，会怎么样？是否还能训练？

我也查看了这个答案，但它没有帮助 question

请谁能解释一下，谢谢。

Answer 1

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

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

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