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 了。