如何在tensorflow中更少地编写昂贵的摘要

时间:2017-04-26 03:18:09

标签: tensorflow

我有张量流模型。在其中,我有不同的摘要。一些,如损失和准确性和便宜,我想经常写它们。其他人,比如测试集上的准确性,计算成本更高,我想写一下,比普通摘要少100倍。在tensorflow中实现它的最佳方法是什么?

1 个答案:

答案 0 :(得分:4)

不是将所有摘要与merge_all()合并,而是使用merge()创建几个不同的摘要组,然后以不同的频率编写它们。像这样:

s1 = tf.summary.image(...)
s2 = tf.summary.scalar(...)
s3 = tf.summary.histogram(...)
s4 = tf.summary.audio(...)

summary_expensive = tf.summary.merge([s1, s4])
summary_cheap     = tf.summary.merge([s2, s3])

# open a session `sess`
# init variables
# create a writer `writer`
for i in xrange(many_steps):
  summary1 = sess.run(summary_cheap)
  writer.add_summary(summary1, i)

  if i % 100 == 0:
     summary2 = sess.run(summary_expensive)
     writer.add_summary(summary2, i)