Keras fit_generator 和 steps_per_epoch

Question

我注意到对于fit_generator，steps_per_epoch 参数通常被分配total_samples//batch_size。

但是，如果我的 total_samples 是 1000 并且我的 batch_size = 32，那么似乎 我应该有 31.25 个批次，但我每个 epoch 只分配 31 个批次。

这是否意味着我将无法训练不完整的批次？

我假设shuffle 选项可以解决这个问题， 但如果我假设一个自定义生成器，shuffle 选项仍然有效吗？

Answer 1

是的，您将在此处仅分配 31 个步骤。你需要做的是：

numpy.ceil(total_samples//batch_size)

由于该模型可以接受任意数量的输入，因此即使最后一批小于 32 也可以工作。