tensorboard中把多次实验的结果绘制在一起

在进行模型的训练时,我们常常会绘制loss,指标或者训练相关的图像,有时候需要将多次实验的结果绘制在一起进行对比。使用tensorborad就可以做到,只需要做到以下两点

  • 不同实验的绘制图像的命名保持一致
  • 将保存好的events文件保存在不同的文件夹,然后加载时定位到父级目录

不同实验的绘制图像的命名保持一致

self.writer.add_scalar( 'Training/Loss', loss_board, self.step)
self.writer.add_scalar( 'Training/Acc', acc_board, self.step)

例如上述添加了loss和acc的曲线,各自的命名为Training/Loss,Training/Acc,那么下一次实验请依旧保持这个命名

将保存好的events文件保存在不同的文件夹,然后加载时定位到父级目录

文件夹示例
文件夹结构
例如上图,在example_1和example_2两个文件夹中都有一个events文件,在用tensorboard打开时定位到上级目录test就可以绘制在一起啦(不要将两个events文件夹放在一起然后打开,这样虽然可以绘制在一起,但是却是错误的曲线)

tensorboard --logdir=./test

Result

重叠曲线

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐