summary.merge_all() 的用法

tf.summary.merge_all 可以將所有summary全部保存到磁盤,以便tensorboard顯示。

例如:

tf.summary.scalar('accuracy',acc)                   #生成準確率標量圖  
merge_summary = tf.summary.merge_all()  
train_writer = tf.summary.FileWriter(dir,sess.graph)#定義一個寫入summary的目標文件,dir爲寫入文件地址  
......(交叉熵、優化器等定義)  
for step in xrange(training_step):                  #訓練循環  
    train_summary = sess.run(merge_summary,feed_dict =  {...})#調用sess.run運行圖,生成一步的訓練過程數據  
    train_writer.add_summary(train_summary,step)#調用train_writer的add_summary方法將訓練過程以及訓練步數保存 

 

生成的流程圖例子:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章