如何使用 tensorboardX 在同一张图表上获得两个标量？

【问题标题】：How to get two scalars on same chart with tensorboardX?如何使用 tensorboardX 在同一张图表上获得两个标量？
【发布时间】：2018-12-02 15:50:29
【问题描述】：

文档似乎表明 add_custom_scalars_multilinechart 可以做到，但它不起作用。

有这样的东西：

from tensorboardX import SummaryWriter
writer = SummaryWriter(comment='test')

writer.add_custom_scalars_multilinechart(['loss/train', 'loss/test'], title='losses')

for blahblah:
    ...
    writer.add_scalar('loss/train', loss.item(), epoch)
    writer.add_scalar('loss/test', loss_test.item(), epoch)

【问题讨论】：

这现在似乎工作了。我不确定 tensorboard 是否处于不良状态或缓存了某些内容。点击周围似乎会触发更改。

标签： python pytorch tensorboardx

【解决方案1】：

使用 tensorboardX 在同一张图表上绘制两个标量：

from tensorboardX import SummaryWriter

为两个标量创建两个 summaryWriter

writer_train = SummaryWriter('runs/train_0')
writer_test = SummaryWriter('runs/test_0')

将标量实例分别添加到summaryWriter；它们必须具有相同的标签，例如：“LOSS”

for data in loop:
    writer_train.add_scalar('LOSS', loss.data.item(), idx)
    writer_test.add_scalar('LOSS', loss_test.data.item(), idx)

工作代码请访问github：Examples with tensorboardX （查看更多_plots_one_chat.py）

教程：TensorboardX

【讨论】：