安装
1.直接pip安装
pip install tensorboardX
2.最新版源码 pip install git+https://github.com/lanpa/tensorboard-pytorch
requirement
- pytorch
- torchvision
- tensorboard
显示
使用时需要先将tensorboard打开：
tensorboard --logdir runs （runs是默认路径）
在浏览器输入：
http://0.0.0.0:6006/
参考资料：
https://github.com/lanpa/tensorboard-pytorch
使用
创建summary writter
在记录任何东西之前，我们需要建立一个writer实体。writer就相当于一个日志，保存你要做图的所有信息。
from tensorboardX import SummaryWriter
writer = SummaryWriter()
不指定参数时，默认路径为runs/CURRENT_DATETIME_HOSTNAME,每次执行程序都会改变。
writer = SummaryWriter()
指定参数，就存入指定路径,如runs/exp-1。
writer = SummaryWriter(‘runs/exp-1’)
还可以指定comment参数，这个参数是一个字符串，这样会把comment加在路径后面,例如下面的调用会得到这样的路径：runs/CURRENT_DATETIME_HOSTNAME_3xLR。
writer = SummaryWriter(comment=‘3xLR’)
创建了wirter之后，接下来要写入任何东西都是调用writer.add_xxx()函数。
API的一般形式
add_something(tag_name, object, iteration_number)
- tag_name: 要写入的数据的名称
- object: 要写入的对象
- iteration_number: 写入的步长，多久写一次
记录标量
写入图像名称，要记录的数值，还有记录的步长（如周期数，或是迭代数）
writer.add_scalar(‘myscalar’, value, iteration)
参考资料
官方文档：https://tensorboard-pytorch.readthedocs.io/en/latest/index.html
github: https://github.com/lanpa/tensorboard-pytorch