tensorboardX的使用

1、首先要安装：

pip install tensorboard (这个过程稍慢，可以选择国内镜像)
pip install tensorboardX

2、编写程序

from tensorboardX import SummaryWriter

# 写入到当前的log文件夹下（先创建一个同级的log文件夹）
writer = SummaryWriter('log')
for i in range(100):
    # a的值是i，步长是i
    writer.add_scalar('a', i, global_step=i)
    writer.add_scalar('b', i**2, global_step=i)

# 关闭读写器
writer.close()

3打开一个终端，进入到log文件夹下执行：

tensorboard --logdir ./

最终效果：