Tensorflow：如何为摘要创建新集合？

Question

我想为摘要创建两个不同的集合。一个是训练总结，一个是验证总结。

所以我可以使用两个不同的 merge_all 操作来存储值

merge_all(key=tf.GraphKeys.SUMMARIES)

函数标量可以添加到集合中。

tf.summary.scalar(
    name,
    tensor,
    collections=None,
    family=None
)

如何为摘要创建新集合？

Answer 1

应该可以使用任意字符串值作为键。它可能看起来像这样：

tf.summary.scalar('tag_a', ...)
tf.summary.scalar('tag_b', ..., collections=["foo"])
merged_a = tf.summary.merge_all()
merged_b = tf.summary.merge_all(key="foo")
writer_a = tf.summary.FileWriter(log_dir + '/collection_a')
writer_b = tf.summary.FileWriter(log_dir + '/collection_b')
for step in range(1000):
  summary_a, summary_b = sess.run([merged_a, merged_b], ...)
  writer_a.add_summary(summary_a, step)
  writer_b.add_summary(summary_b, step)

值得一提的是，通常人们会以一种方式配置事物，即有一个 merge_all 操作和多个 run 调用。例如：https://github.com/tensorflow/tensorflow/blob/cf7c008ab150ac8e5edb3ed053d38b2919699796/tensorflow/examples/tutorials/mnist/mnist_with_summaries.py#L142 即使使用 collections 参数分解汇总系列，TensorBoard 仍会在视觉上表示它们，就好像它们是不同的运行一样。另请注意，name 参数对应于运行中的每个图表（标签）。