logging - 如何在TensorFlow摘要中添加任意值？

Question

为了将一个简单的val值记录到TensorBoard摘要中，我需要

val = 5
test_writer.add_summary(sess.run(tf.scalar_summary('test', val)), global_step)

是

sess.run(tf.scalar_summary('test', val))

真的有必要将val添加为摘要吗？

Answer 1

您可以自己构建摘要，例如

from tensorflow.core.framework import summary_pb2

value = summary_pb2.Summary.Value(tag="Accuracy", simple_value=0.95)
summary = summary_pb2.Summary(value=[value])

然后，您可以像代码中一样使用summary添加add_summary。