tensorboardX的使用

环境使用pytorch,以及tensorboardX

例子

from tensorboardX import SummaryWriter

#tensorboard --logdir="test/data/chap4/log"
SummaryWriter=SummaryWriter(logdir="data/chap4/log")

glob_step=1
for epoch in range(5):
    for step in range(10):
        SummaryWriter.add_scalar("train loss",glob_step+0.1,glob_step)
        glob_step = glob_step + 1

启动命令

tensorboard --logdir="test/data/chap4/log"

 结果