从 tf.train.summary_iterator 中检索全局步数

Question

我正在使用 tf.train.summary_iterator 读取事件文件中存储的数据。我想访问存储特定摘要的全局步骤编号，但 summary.proto 没有引用步骤编号。 Tensorboard 似乎可以访问这些信息，所以我认为可能有办法做到这一点，但我一直无法找到它。

Answer 1

tf.train.summary_iterator 正在遍历 Event 个对象：

An iterator for reading Event protocol buffers from an event file.

... 其中包含该事件的全局步数（参见 event.proto）。尝试：

for e in tf.train.summary_iterator(path to events file):
  print(e.step)

Retrieve global step number from tf.train.summary_iterator