Tensorboard:绘制多个模型的损失曲线

时间:2017-09-13 22:35:27

标签: tensorflow tensorboard

在张量板中,很容易显示单个CNN模型的损耗曲线。如果我单独训练了两个模型(例如inception-v4和ResNet)并且我想同时绘制那些模型的损耗曲线(如下图所示),我该怎么办? enter image description here

1 个答案:

答案 0 :(得分:0)

我认为您可以通过创建两个单独的FileWriter来实现这一目标 - 每个模型一个:

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

然后使用这些编写器分别为inpection和ResNet模型添加摘要。

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

唯一要做的就是使用
来运行tensorBoard  命令 tensorboard --logdir=/tmp/TensorBoards/example TensorBoard会自动将数据合并到一个图中

就我而言,它看起来像this

BTW有一个谷歌开发者提供的基本张量板功能的真实有趣的演示 https://www.youtube.com/watch?v=eBbEDRsCmv4&t=773s