tensorboardX

安装

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