我有两个不同的摘要组。一个在每一批次中收集一次,另一个在每一轮中收集一次。我如何使用merge_all_summaries(key='???')
来分别收集这两个组中的摘要?手动执行是一种选择,但似乎有更好的方法。
以下是我认为它应该如何工作的示例:
# once per batch
tf.scalar_summary("loss", graph.loss)
tf.scalar_summary("batch_acc", batch_accuracy)
# once per epoch
gradients = tf.gradients(graph.loss, [W, D])
tf.histogram_summary("embedding/W", W, collections='per_epoch')
tf.histogram_summary("embedding/D", D, collections='per_epoch')
tf.merge_all_summaries() # -> (MergeSummary...) :)
tf.merge_all_summaries(key='per_epoch') # -> NONE :(