tensorboardX是pytorch集成tensorboard可视化训练过程的工具。
绘制loss曲线的方法如下
- 单条曲线使用
writer.add_scalar，一条曲线对应一个name - 多条曲线使用
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()
-
单条曲线
acc
-
多条曲线
train/val loss
网友评论