如何在日志文件中添加选择性节点

时间:2017-12-08 21:35:48

标签: tensorflow tensorboard

我正在尝试对训练有素的模型进行测试。我已在会话中恢复了检查点。该模型有很多操作,但我只测试其中的一部分。

我希望看到我在张量板上测试的op网络,但是目前,它都与我不想要的所有其他操作纠缠在一起。

有没有办法存储您在会话中执行的选择性操作的tf.summary

1 个答案:

答案 0 :(得分:1)

您可以合并您想要的操作merged_summary = tf.summary.merge_all(),而不是按merged_summary_group1 = tf.summary.merge([op1, op2, ...])合并所有摘要。之后,将merged_summary中的所有sess.run替换为merged_summary_group1