PyTorch 可视化工具 TensorboardX

 

可视化教程

• • 使用步骤
  • • 1. 导入包
    • 2. 添加标量、模型图
  • 所遇到问题
  • • 只显示 scalar，不显示 graph 解决办法

 

使用步骤

1. 导入包

from tensorboardX import SummaryWriter

2. 添加标量、模型图

with SummaryWriter() as w:
    w.add_scalar('scalar/test', loss.item(), (i_batch+1+200*epoch) )
    w.add_scalar('scalar/epoch', loss.item(), (i_batch+1+200*epoch) )

显示，在终端中输入以下指令

tensorboard --logdir==runs

tensorboard --logdir==runs --port 6006

tensorboard --logdir==r'D:\board\logs'

with SummaryWriter() as w:
    w.add_graph(AnNet, (img, ))

参考

1. Pytorch的网络结构可视化（tensorboardX）（详细）
2. Pytorch使用tensorboardX可视化。超详细！！！
3. https://github.com/lanpa/tensorboardX

所遇到问题

只显示 scalar，不显示 graph 解决办法

1. Win10中pyTorch1.4.0+tensorboard配置后graph显示空白
2. pytorch1.4+tensorboard不显示graph计算图的问题
3. 【pytorch】pytorch 1.2中tensorboard无法显示graph解决办法
4. 解决TensorBoard看不到Graphs
5. No dashboards are active for the current data set. 解决pytorch下tensorboardx无法显示

一直各种尝试，最终终于解决了。我把 anaconda 重新卸载了，然后全局搜索了下文件夹，确保每个地方都没有了 conda 的东西。因为之前安装和删除包的时候分别使用了 conda install(/remove) XXX 和 pip install(/uninstall) XXX 命令，导致 tensorboard 这个包重复安装了，且版本错乱。

最后重装 anaconda 之后，仍然不起作用。卸载了 tensorflow ，单独安装 tensorboard 包之后发现，终于，正常显示 graph 了。

正式使用时的版本如下：

之后为了再次验证 TensorFlow 包安装时无法使用的问题，再次使用 pip install tensorflow-gpu 指令下载安装 TensorFlow 包。

好吧~~~ 安装之后依然可以正常显示，并没有影响。

不过我发现一个有趣的现象，就是我的 tensorboard 打开的时候，每次下载 TensorFlow 总是会断掉，之前以为是网络不稳定，会自己切换造成的，之后我把所有的无线网络全部关掉，只留一个网络，还是一样的效果。反而把 tensorboard 服务关闭之后，便可以正常下载 TensorFlow 了。