tensorboardX

安装

1.直接pip安装
pip install tensorboardX

2.最新版源码 pip install git+https://github.com/lanpa/tensorboard-pytorch

requirement

  • pytorch
  • torchvision
  • tensorboard

显示

使用时需要先将tensorboard打开:

tensorboard --logdir runsruns是默认路径)

在浏览器输入:
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

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容