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

How to add selective nodes in the log file

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

我想在 tensorboard 中看到我正在测试的 op 网络,但目前,它都与我不想要的所有其他操作纠缠在一起。

有什么方法可以存储您在会话中执行的选择性操作的 tf.summary 吗?

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