在做網(wǎng)絡(luò)訓(xùn)練實(shí)驗(yàn)時(shí)，有時(shí)需要同時(shí)將訓(xùn)練曲線和測(cè)試曲線一起顯示，便于觀察網(wǎng)絡(luò)訓(xùn)練效果。經(jīng)過很多次踩坑后，終于解決了。
具體的方法是：設(shè)置兩個(gè)writer，一個(gè)用于寫訓(xùn)練的數(shù)據(jù)，一個(gè)用于寫測(cè)試數(shù)據(jù)，并且這兩個(gè)writer分別存在train和test路徑中，注意測(cè)試的writer不能加sess.graph如下代碼所示。
|
1
2
3
4
5
6
7
8
9
10
|
...train_log_dir = 'logs/train/'test_log_dir = 'logs/test/' # 兩者路徑不同megred = tf.summary.merge_all()with tf.Session() as sess: writer_train = tf.summary.FileWriter(train_log_dir,sess.graph) writer_test = tf.summary.FileWriter(test_log_dir) # 注意此處不需要sess.graph ...other code... writer_train.add_summary(summary_str_train,step) writer_test.add_summary(summary_str_test,step) |
此處貼一個(gè)使用mnist測(cè)試的圖片，如圖所示，在左下角可以通過test和train的選項(xiàng)卡選擇查看哪條曲線。
以上這篇tensorboard實(shí)現(xiàn)同時(shí)顯示訓(xùn)練曲線和測(cè)試曲線就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持服務(wù)器之家。
原文鏈接：https://blog.csdn.net/shahuzi/article/details/81223318
