How to use several summary collections in Tensorflow?

前端 未结 1 1257
自闭症患者
自闭症患者 2020-12-24 15:26

I have 2 distinctive groups of summaries. One is collected once per batch another one is collected once per epoch. How can I use merge_all_summaries(key=\'???\')

相关标签:
1条回答
  • 2020-12-24 15:45

    Problem solved. collections parameter of a summary is supposed to be a list. Solution:

      # once per batch 
      tf.scalar_summary("loss", graph.loss)
      tf.scalar_summary("batch_acc", batch_accuracy)
      # once per epoch
      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') # -> (MergeSummary...) :)
    

    Edit. Syntactical change in TF:

    # once per batch 
      tf.summary.scalar("loss", graph.loss)
      tf.summary.scalar("batch_acc", batch_accuracy)
      # once per epoch
      tf.summary.histogram("embedding/W", W, collections=['per_epoch'])
      tf.summary.histogram("embedding/D", D, collections=['per_epoch'])
    
      tf.summary.merge_all()                # -> (MergeSummary...) :)
      tf.summary.merge_all(key='per_epoch') # -> (MergeSummary...) :)
    
    0 讨论(0)
提交回复
热议问题