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