tensorflow - 从 Tensorflow 中的 .tfrecords 文件获取记录总数

Question

是否可以从 .tfrecords 中获取记录总数？文件 ？与此相关的是，人们通常如何跟踪训练模型时已经过去的 epoch 数？虽然我们可以指定 batch_size和 num_of_epochs ，我不确定获取诸如 current epoch 之类的值是否很简单。 ，每个时期的批次数等 - 只是为了让我可以更好地控制培训的进展情况。目前，我只是使用一个肮脏的黑客来计算这个，因为我事先知道我的 .tfrecords 文件中有多少记录以及我的小批量的大小。感谢任何帮助..

Answer 1

要计算记录数，您应该可以使用 tf.python_io.tf_record_iterator .

c = 0
for fn in tf_records_filenames:
  for record in tf.python_io.tf_record_iterator(fn):
     c += 1

为了跟踪模型训练， tensorboard派上用场了。