如何将多个摘要编写器与 TensorBoard 的 `tf.train.Supervisor` 一起使用答案

【问题标题】：How to use multiple summary writers with `tf.train.Supervisor` for TensorBoard如何将多个摘要编写器与 TensorBoard 的 `tf.train.Supervisor` 一起使用
【发布时间】：2017-01-30 14:28:30
【问题描述】：

我想做类似于TensorBoard tutorial 中的train_writer 和test_writer 的事情。但是使用tf.train.Supervisor。但是，我不确定如何最好地解决这个问题。

伪代码：

train_op = #...
train_summaries = # ...
test_summaries = # ...

config = tf.ConfigProto(allow_soft_placement=True)
sv = tf.train.Supervisor(
    logdir              = ????,
    summary_op          = ????,
    summary_writer      = ????,
)
with sv.managed_session(config=config) as sess:
    while not sv.should_stop():
        sess.run(train_op)

所以我的问题是：如何保存train_summaries 和test_summaries 做不同的目录？ ./logdir/train 和 ./logdir/test/。

【问题讨论】：

标签： tensorflow deep-learning tensorboard

【解决方案1】：

您正在寻找summary_computed。文档字符串显示了如何创建自定义摘要编写器。您无法让主管自动管理它，但这很简单。来自https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/training/supervisor.py

# Create a Supervisor with no automatic summaries.
sv = Supervisor(logdir='/tmp/mydir', is_chief=is_chief, summary_op=None)
# As summary_op was None, managed_session() does not start the
# summary thread.
with sv.managed_session(FLAGS.master) as sess:
  for step in xrange(1000000):
    if sv.should_stop():
      break
    if is_chief and step % 100 == 0:
      # Create the summary every 100 chief steps.
      sv.summary_computed(sess, sess.run(my_summary_op))
    else:
      # Train normally
      sess.run(my_train_op)

【讨论】：