原文地址

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. 单条曲线

  1. 多条曲线