Tensorboard：一起绘制多个模型的损失曲线

Question

在tensorboard中，很容易显示单个CNN模型的损失曲线。如果我分别训练了两个模型（例如inception-v4和ResNet）并且我想同时绘制它们的损失曲线（如下图），我该怎么办？ enter image description here

Answer 1

我认为您可以通过创建两个单独的 FileWriters 来实现这一点 - 每个模型一个：

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

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

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

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

在我的例子中它看起来像 this

顺便说一句，google 的一位开发人员展示了非常有趣的基本 tensorboard 功能演示 https://www.youtube.com/watch?v=eBbEDRsCmv4&t=773s

Tensorboard：一起绘制多个模型的损失曲线

Tensorboard: Plot loss curves of mutiple models together

tensorflow

tensorboard