Tensorflow:要知道一个epoch是否在迭代中完成

C. *_*ang 5 tensorflow

我使用 Tensorflow 的队列作为训练网络的输入。我设置num_epoches为10和batch_size100。由于我不知道TFRecords文件中存储了多少样本,因此我不知道一个epoch完成了多少次迭代。

我知道有一种方法可以tf.python_io.tf_record_iterator用来预读 TFRecords 文件,以获得样本总数。但是当 TFRecords 文件很大时,这种方法是不好的,因为它必须循环整个文件并计算样本。我想知道Tensorflow是否有自己的方式报告一个epoch结束?