WARNING:tensorflow:Your input ran out of data； interrupting training. 解决方法-Toy模板网

这篇具有很好参考价值的文章主要介绍了WARNING:tensorflow:Your input ran out of data； interrupting training. 解决方法。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

问题详情：

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, 13800 batches). You may need to use the repeat() function when building your dataset.

根据报错信息，问题似乎出现在数据集生成器上。可以使用数据集的repeat()函数来确保生成足够的批次供训练使用。

我们可以对训练数据集（train_dataset）应用repeat()函数，以确保生成足够的批次：

train_dataset = train_dataset.repeat()

问题原因：

generator产生的训练数据不够用，少于所要求的steps_per_epoch * epochs 个batch数。
(len(train_data)/batch_size)*epochs=所需要的训练数据的batch数
如果盲目的按照提示repeat()函数来扩充train_data的话，要求的所需的batch数也会相应地增多，会陷入一条死路。。

具体修改如下：

在我的代码中，在定义好的各种函数下，在steps_per_epoch = len(X_train) // BATCH_SIZE之前。

WARNING:tensorflow:Your input ran out of data； interrupting training. 解决方法,问题解决,tensorflow,人工智能,python

train_dataset = build_dataset(X_train, Y_train, bsize=BATCH_SIZE, decode_fn=train_decoder, 
                            augmentAdv=False, augment=False, augmentAdvSeg=True)
train_dataset = train_dataset.repeat()

这样，在开始训练之前，数据集将使用 repeat() 函数来确保生成足够的批次供训练使用。

之前两个epoch就会停止，现在可以正常运行了

WARNING:tensorflow:Your input ran out of data； interrupting training. 解决方法,问题解决,tensorflow,人工智能,python 文章来源地址https://www.toymoban.com/news/detail-729283.html

到了这里，关于WARNING:tensorflow:Your input ran out of data； interrupting training. 解决方法的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！