2017-09-13 19 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

非常感谢您的回复!我会基于此尝试。 –