日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

pytorch?tensorboard可視化的使用詳解_python

作者:Caesar6666 ? 更新時間: 2022-11-23 編程語言

一、 安裝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“,關閉端口。

原文鏈接:https://blog.csdn.net/Caesar6666/article/details/127245418

欄目分類
最近更新