我正在使用 Tensorboard 来可视化 Tensorflow 的运行,并且我想要一个摘要图,每个 epoch 只写入一次值。
我想做这样的事情:
with graph.as_default():
tf_ending = tf.placeholder(tf.bool)
tf.scalar_summary('Loss', loss) # Some summaries are written every time
if tf_ending:
# This summary should only get written sometimes.
tf.scalar_summary('Total for Epoch', epoch_total)
我觉得我需要做一些
tf.merge_all_summaries()
以外的事情并分别管理摘要集,但我不确定这将如何工作。
最佳答案
一种方法是向 SummaryWriter 添加自定义的 Summary protobuf。在每个纪元结束时( session /图表之外),您可以添加如下内容:
summary = tf.Summary()
summary.value.add(tag='Total for Epoch',simple_value=epoch_total)
summary_writer.add_summary(summary, train_step)
然而,这需要通过 tensorflow 图(sess.run)返回值(epoch_total)。另外,我不确定这是否是执行此类操作的最佳方法,但是您确实会在 TF 示例中看到此方法,例如here和 here .
关于Tensorflow - 有条件地向 tensorboard 写入摘要,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40052321/