首頁 > 軟體

pytorch tensorboard視覺化的使用詳解

2022-10-14 14:00:03

一、 安裝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視覺化的使用詳解的文章就介紹到這了,更多相關pytorch tensorboard內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com