Tensorboard 雖然是 tensorflow 內置的可視化工具,但是他們跑在不同的進程中,所以 可以將 tensorboard 應用到 Pytorch 中,可以幫助我們理解整個神經網絡的學習過程、數據的分佈、性能瓶頸等等。
那 Tensorboard 這麼好用,咱肯定要試試看,但安裝完成後 打開 cmd 終端,輸入 tensorboard --logdir logs 命令啓動,複製 http://localhost:6006/ 在瀏覽器中打開即可看到Web頁面,但是卻顯示 No Image data was found (如下圖):
折騰了好久,纔將這個問題解決掉,現將其總結如下:
(1)、查看 logs 的文件夾的地址;E:\Jupyter_Notebook Project\PyTorch basic
以我爲例:E:\Jupyter_Notebook Project\PyTorch basic
(2)、修改 cmd 的默認位置,即:
cd /d E:\Jupyter_Notebook Project\PyTorch basic
(3)、輸入 Tensorboard 啓動命令;
tensorboard --logdir=logs
(4)、將 http://localhost:6006/ 複製到 Google 瀏覽器打開就能看到數據。