deep-learning - Keras fit_generator 和 steps_per_epoch

Question

我注意到 fit_generator ，steps_per_epoch 参数通常分配total_samples//batch_size .

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

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

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

Answer 1

是的，您在这里只会被分配 31 个步骤。你需要做的是:

numpy.ceil(total_samples//batch_size)

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