python - TensorFlow:如何合并多个 'collections' ？

Question

我有一些集合，我想使用主管使用 TensorBoard 进行跟踪。在 Supervisor 初始化程序中，我想要一些效果

summary_op = tf.summary.merge_all(['test', 'valid'])

但我收到错误 TypeError: unhashable type: 'list'，因为 key 必须是字符串 ( see documentation )。

---

编辑:

这也行不通:

summary_op = [tf.summary.merge_all('train'), tf.summary.merge_all('valid')]

Answer 1

试试 tf.summary.merge()，例如像这样:

summary_op = tf.summary.merge([
        tf.summary.merge_all('test'),
        tf.summary.merge_all('train')],
    collections='merged')

这将合并来自test 和train 集合的所有摘要，并将它们添加到新的merged 集合中。请记住，如果在相同时间步内多次使用相同摘要，这将导致奇怪的效果:

在这里，我不小心(手动!)在训练运行期间存储了验证摘要，然后再次在单独的验证运行中。

此外，我不确定这是否是最有效的方法，但它确实有效。