tf.data + generator + keras => repeat()不起作用,为什么?

问题描述 投票:0回答:1

[当我训练我的数据集(tf.data.dataset.from_generator(x),其中x是由]创建的迭代器时>

df.rdd.flatMap(func).toLocalIterator() 

将func应用于我的数据集。 repeat()函数不适用于数据集。我正在使用tensorflow 1.14.0版本。错误是:

您的数据集迭代器用完了数据;中断训练。使确保您的迭代器至少可以生成validation_steps * epochs个批次(在这种情况下为3个批次)。您可能需要使用构建数据集时使用repeat()函数。

使重复功能起作用的可能解决方法是什么?我可以提供更多细节,但是我认为正在发生的只是该功能本身在我的条件下不起作用。

[当我训练我的数据集(tf.data.dataset.from_generator(x)时,使用x df.rdd.flatMap(func).toLocalIterator()创建的迭代器,其中func应用于我的数据集。repeat()函数是否...

tensorflow keras generator tensorflow-datasets training-data
1个回答
0
投票
© www.soinside.com 2019 - 2024. All rights reserved.