如何只通过一次tf.data.dataset？

Question

我有一个tfrecord数据集。我想对数据集进行推断，因此我只想使用一次数据集。但是，如果我继续运行迭代器，最终它将抛出超出范围的错误。

Answer 1

典型的处理方法是捕获OutOfRangeError：

...  # set up model, data etc.
next_batch_op = iter.get_next()

try:
    while True:
        batch = sess.run(next_batch_op)
        ...  # do something with batch, e.g. inference
except tf.errors.OutOfRangeError:
    ... # finalize the program

这将在数据集中运行一次，一旦抛出错误，您可以执行其他操作。