TensorBoard---------tensorboardX版

from tensorboardX import SummaryWriter

# 实例化SummaryWriter，并指明日志存放路径(若当前目录没有此文件将自动创建)
writer = SummaryWriter(log_dir='logs')
# 调用对应的实例
writer.add_xxx()
# 关闭writer
writer.close()

# 启动tensorboard服务
tensorboard --logdir=logs --port localhost

举例

import torch
from tensorboardX import SummaryWriter
writer = SummaryWriter()

for n_iter in range(10):
    scalar = torch.rand(1)
    writer.add_scalar('data/scalar',scalar[0],n_iter)
    writer.add_scalar('data/scalar_group',{'sinx':torch.sin(x)[0],
                                           'cosx':torch.cos(x)[0],
                                           'atanx':torch.atan(x)[0]},n_iter)
writer.close()