pytorch?tensorboard可視化如何使用

今天小編給大家分享一下pytorch tensorboard可視化如何使用的相關知識點，內容詳細，邏輯清晰，相信大部分人都還太了解這方面的知識，所以分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后有所收獲，下面我們一起來了解一下吧。

一、 安裝tensorboard

直接pip即可:

pip install tensorboard

這里注意，使用pytorch，并不需要額外安裝tensorflow。

二、 使用tensorboard

記錄訓練的loss和測試的accuracy：

1、首先導入模塊

from torch.utils.tensorboard import SummaryWriter

2、初始化

writer = SummaryWriter('./log')

記錄的內容保存在文件夾log里面。

3、記錄內容

writer.add_scalar('loss', loss, step)
writer.add_scalar('accuracy', accuracy, step)

這里step是指步數，loss為訓練誤差，accuracy為測試準確率，按自己需求記錄即可。

4、關閉

writer.close()

在文件夾會log里會生成events.out.tfevents的文件，每次運行，如：

三、可視化

1、打開終端寫命令行

在本地打開終端，進入虛擬環境，寫以下命令行：

tensorboard --logdir=./log

這里需要注意一下，log為剛才保存文件的文件夾，可以是相對路徑，也可以是絕對路徑，只要稍微注意格式即可，以下有幾個注意事項：

（1）、路徑./log不要加雙引號,，如：–logdir=“./log”

（2）、不要使用雙等號，有些博客使用了雙等號，其實是錯誤的，如：–logdir==./log

以上寫法均是錯誤的，正確的寫法有兩種，一種就是上面例子使用單個等號，另外一種是使用空格，如：

tensorboard --logdir ./log

參數除了logdir外，常用的還有：host、port。

host：ip地址

port: 端口

tensorboard的默認是：

tensorboard --logdir=./log --host=127.0.0.1 --port=6006

有些博客說預防端口被占用，常常修改端口，其實一般不需要，host和port默認就好，只需寫logdir就好，即按照開頭那樣寫即可，回車后，有：

注意：這里需要注意一下，不要著急按下”Ctrl+C“，不然無法打開http://localhost:6006/，當不需要打開http://localhost:6006/，才按下”Ctrl+C“，即斷開端口。

2、打開瀏覽器打開tensorboard可視化

當不需要可視化了，可以去終端按下”Ctrl+C“，關閉端口。

以上就是“pytorch tensorboard可視化如何使用”這篇文章的所有內容，感謝各位的閱讀！相信大家閱讀完這篇文章都有很大的收獲，小編每天都會為大家更新不同的知識，如果還想學習更多的知識，請關注億速云行業資訊頻道。