溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

tensorboard實(shí)現(xiàn)同時(shí)顯示訓(xùn)練曲線和測(cè)試曲線

發(fā)布時(shí)間：2020-10-24 14:08:02 來源：腳本之家 閱讀：290 作者：shahuzi 欄目：開發(fā)技術(shù)

在做網(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如下代碼所示。

...
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è)試曲線

以上這篇tensorboard實(shí)現(xiàn)同時(shí)顯示訓(xùn)練曲線和測(cè)試曲線就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手