从Tensorflow 1.8.0升级到1.11.0后，在每个时期记录OutOfRangeError

Question

我有一个TfRecordDataset：

...
dataset = dataset.batch(batch_size)
dataset = dataset.prefetch(buffer_size=tf.contrib.data.AUTOTUNE)
dataset_iterator = dataset.make_initializable_iterator()
(...) = dataset_iterator.get_next()

然后是标准：

test_session.run(dataset_iterator.initializer)
while not done:
    try:
        ... = test_session.run([...])
        batch_count += 1
    except tf.errors.OutOfRangeError:
        done = True

这与Tensorflow 1.8.0完美配合。今天我将Tensorflow升级到TF 11.1.0，现在我有一个错误：OutOfRangeError: End of sequence [[{{node dataset/IteratorGetNext}} = IteratorGetNext

在一个时代的每一端。实际上已捕获错误本身，但错误将打印在控制台上。有人可以指出实际发生了什么吗？

Answer 1

问题在TF 1.12.0中得到解决。