PyTorch中使用TensorboardX

1. Tensorboard 的安装

TensorboardX是在tensorboard的基础上做出来。因此很多文章[1][2]认为,它需要tensorflow支持,使用时应当安装TensorFlow。

根据其github的介绍,也是需要安装TensorFlow。不过,我们不一定要在pytorch环境中安装TensorFlow,这样会显得很混乱。不安装TensorFlow时,执行:

tensorboard --logdir=路径名

此时可能会报错:

ImportError: Something is wrong with the numpy installation. 
While importing we detected an older version of numpy in ['/usr/local/lib/python2.7/dist-packages/numpy']. 
One method of fixing this is to repeatedly uninstall numpy until none is found, then reinstall this version.

实际上和numpy没有关系 ,它是主动去调用base 环境里面已经安装的TensorFlow中的tensorboard。实际上,我们可以只用在pytorch所在环境里安装tensorboard即可:

pip install tensorboard
pip install tensorboardX

有的时候网址可能打不开,只需要正常打开谷歌浏览器,输入http://localhost:6006即可正常打开tensorboard。这是由于版本不同导致的。

补充说明
PyTorch 1.4版本内置了tensorboard,支持​add_graph ​,可以将定义的模型的计算图可视化出来,直观地看到每一层的size和数据流向,为模型调试和验证提供帮助。
参见文档: https://pytorch.org/docs/stable/tensorboard.html

2. TensorBoard 的使用

TensorBoard 的使用,【TensorFlow | TensorBoard】理解 TensorBoard 这篇文章写得很清楚。现就本人实际使用中遇到的问题进行总结。

- TensorBoard 中 SCALAS里面有些曲线没有显示

其中每个图的右下角都有 3 个小图标,第一个是查看大图,第二个是是否对 y 轴对数化,第三个是如果你拖动或者缩放了坐标轴,再重新回到原始位置。图源见水印。
在这里插入图片描述
但是有的时候调整这三个选项都无法调出曲线来,这种情况多发生在训练前几步中,原因是选中了页面左上的 Ignore outliers in chart scaling 选项:

页面左上有两个选项,分别是 Show data download linksIgnore outliers in chart scaling,第一个显示数据下载链接,可以把 TensorBoard 作图用的数据下载下来,点击后可以在图的右下角可以看到下载链接以及选择下载哪一个 run 的,下载格式支持 CSV 和 JSON。第二个是排除异常点,默认选中。该选项选中前后的区别如下:
在这里插入图片描述
在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值