2017-09-13 18 views
0

在張量板中,很容易顯示單個CNN模型的損耗曲線。如果我單獨訓練了兩個模型(例如inception-v4和ResNet),並且我想同時繪製這些模型的損失曲線(如下圖所示),我應該怎麼做? enter image description here張量板:將多個模型的損耗曲線一起繪製

+0

我也想知道這個答案。當然,您可以將損失顯式地保存到日誌文件中,然後編寫一段腳本以便稍後提取並繪製曲線,但將它們結合到TensorBoard中會非常方便。例如,另一個不錯的使用案例會將培訓和驗證損失放在同一個地塊上。 –

+0

謝謝你的回覆。我想你可能知道解決這個問題的方法。你如何「寫一個腳本來提取和繪製曲線」?假設我有兩個單獨的日誌文件。 –

+0

只需從損失計算中將打印輸出添加到日誌文件中,或者打開另一個文件並將其添加到那裏,如果要控制與記錄器分開的格式(如果使用'logging',則意味着您有一堆當需要處理的時候需要處理這些問題)。 –

回答

0

我想你可以通過創建兩個單獨的FileWriter引擎實現這一點 - 一個爲每個模型:利用這些作家分別添加了inpection和RESNET模型摘要

inceptionWriter = tf.summary.FileWriter('/tmp/TensorBoards/example/inception4') 
ResnetWriter = tf.summary.FileWriter('/tmp/TensorBoards/example/ResNet') 

然後。

inceptionWriter.add_summary(loss_summary, step) 
* 
* 
* 
ResnetWriter.add_summary(loss_summary, step) 

唯一剩下要做的就是使用
命令 tensorboard --logdir=/tmp/TensorBoards/example TensorBoard會自動合併數據在一個情節

對我來說,它看起來運行tensorBoard像this

BTW有由谷歌開發人員提供的基本張力板功能真正有趣的演示 https://www.youtube.com/watch?v=eBbEDRsCmv4&t=773s

+0

非常感謝您的回覆!我會基於此嘗試。 –