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 links
和 Ignore outliers in chart scaling
,第一个显示数据下载链接,可以把 TensorBoard 作图用的数据下载下来,点击后可以在图的右下角可以看到下载链接以及选择下载哪一个 run 的,下载格式支持 CSV 和 JSON。第二个是排除异常点,默认选中。该选项选中前后的区别如下: