这些文档似乎表明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)发布于 2019-01-14 21:15:34
用tensorboardX在同一图表上绘制两个标量:
from tensorboardX import SummaryWriter为两个标量创建两个summaryWriter
writer_train = SummaryWriter('runs/train_0')
writer_test = SummaryWriter('runs/test_0')将标量实例添加到相应的summaryWriter;它们必须具有相同的标记,例如:“丢失”
for data in loop:
writer_train.add_scalar('LOSS', loss.data.item(), idx)
writer_test.add_scalar('LOSS', loss_test.data.item(), idx)有关工作代码,请访问github:使用tensorboardX的示例 (参见more_plots_one_chat.py)
教程:TensorboardX
https://stackoverflow.com/questions/53581904
复制相似问题