Keras 序列、fit_generator 和 steps_per_epoch

Question

我注意到对于fit_generator，steps_per_epoch 参数通常被分配total_samples//batch_size，在这里可以创建一个生成器/使用ImageDataGenerator 并将其作为参数传递给fit_generator。

但是我使用 Sequence 类 (keras.utils.Sequence()) 来创建我的生成器并传递 steps_per_epoch 一个小于 total_samples//batch_size 的整数。

我想知道的是，一旦每个 epoch 完成，生成器中的数据生成是否会从头开始？

例如，我的训练集中有 3200 个样本，我使用的批量大小为 32。因此，理想情况下，对于一个完整的 epoch，我应该将 steps_per_epoch 设置为 100。但是，如果我将 steps_per_epoch 设置为50？一旦第一个 epoch 完成，将生成数据点编号 1601 (32*50) 还是从头开始（数据点编号 1）？

Answer 1

使用Sequence时，不需要传递steps_per_epoch，因为这个信息可以从你的Sequence.的__len__方法推断出来

如果你在使用Sequence 时传递steps_per_epoch，这将覆盖__len__ 方法的任何使用，它实际上只会使用序列中的steps_per_epoch 样本（从0 到steps_per_epoch - 1），并且它将在 epoch 结束时重置为零。您可以在keras source code 中检查此行为。