TensorBoardX绘制曲线的简单例子

tensorboardX是pytorch集成tensorboard可视化训练过程的工具。

绘制loss曲线的方法如下

1. 单条曲线使用writer.add_scalar，一条曲线对应一个name
2. 多条曲线使用writer.add_scalars，每个表格对应一个name，其中每条曲线也对应一个name

Example

from tensorboardX import SummaryWriter
import random
import time

writer = SummaryWriter()

for i in range(1000):
    for x in ['train', 'val']:
        writer.add_scalars("loss", {
            x: random.random()
        }, i)
    writer.add_scalar("acc", random.random(), i)
    time.sleep(1)

writer.close()

1. 单条曲线

   
   acc

2. 多条曲线

   
   train/val loss