您的输入数据已用完;中断训练.确保您的数据集或生成器至少可以生成"steps_per_epoch*划时代&qu

Question

我正在尝试使用Keras训练神经网络模型，需要帮助解决这个错误，当我尝试运行我的训练模型时会出现这个错误：

WARNING:tensorflow:Your input ran out of data; interrupting training. Make sure that your dataset or generator can generate at least `steps_per_epoch * epochs` batches (in this case, 19990 batches). You may need to use the repeat() function when building your dataset.

代码可以在这里找到：

https://colab.research.google.com/drive/1hgQoo7Oz6UPIzQQ4ELwRUuPPNxaOAqeh?usp=sharing

Answer 1

我建议您使用fit()方法，而不是像TensorFlow已经建议的那样使用fit_generator()方法。

然而，回到您的问题，实际导致问题的原因是，当使用生成器时，您应该让模型计算出一个历元需要多少步骤才能完成如果手动设置steps_per_epoch，则需要自己正确计算(steps_per_epoch=(data_samples/batch_size)(，如果计算错误，则会发生此错误(在您的情况下就是这样(。因此，为了防止这个问题，您应该在调用fit_generator()方法时删除step_per_epoch参数，让模型自己处理step_per_epoch值，这样代码就可以正常工作了。