pytorch tensorboard可視化的使用詳解
一、 安裝tensorboard
直接pip即可:
pip install tensorboard
這里注意,使用pytorch,并不需要額外安裝tensorflow。
二、 使用tensorboard
記錄訓(xùn)練的loss和測(cè)試的accuracy:
1、首先導(dǎo)入模塊
from torch.utils.tensorboard import SummaryWriter
2、初始化
writer = SummaryWriter('./log')記錄的內(nèi)容保存在文件夾log里面。
3、記錄內(nèi)容
writer.add_scalar('loss', loss, step)
writer.add_scalar('accuracy', accuracy, step)
這里step是指步數(shù),loss為訓(xùn)練誤差,accuracy為測(cè)試準(zhǔn)確率,按自己需求記錄即可。
4、關(guān)閉
writer.close()
在文件夾會(huì)log里會(huì)生成events.out.tfevents的文件,每次運(yùn)行,如:

三、可視化
1、打開終端寫命令行
在本地打開終端,進(jìn)入虛擬環(huán)境,寫以下命令行:
tensorboard --logdir=./log
這里需要注意一下,log為剛才保存文件的文件夾,可以是相對(duì)路徑,也可以是絕對(duì)路徑,只要稍微注意格式即可,以下有幾個(gè)注意事項(xiàng):
(1)、路徑./log不要加雙引號(hào),,如:–logdir=“./log”
(2)、不要使用雙等號(hào),有些博客使用了雙等號(hào),其實(shí)是錯(cuò)誤的,如:–logdir==./log
以上寫法均是錯(cuò)誤的,正確的寫法有兩種,一種就是上面例子使用單個(gè)等號(hào),另外一種是使用空格,如:
tensorboard --logdir ./log
參數(shù)除了logdir外,常用的還有:host、port。
host:ip地址
port: 端口
tensorboard的默認(rèn)是:
tensorboard --logdir=./log --host=127.0.0.1 --port=6006
有些博客說預(yù)防端口被占用,常常修改端口,其實(shí)一般不需要,host和port默認(rèn)就好,只需寫logdir就好,即按照開頭那樣寫即可,回車后,有:

注意:這里需要注意一下,不要著急按下”Ctrl+C“,不然無法打開http://localhost:6006/,當(dāng)不需要打開http://localhost:6006/,才按下”Ctrl+C“,即斷開端口。
2、打開瀏覽器打開tensorboard可視化

當(dāng)不需要可視化了,可以去終端按下”Ctrl+C“,關(guān)閉端口。
到此這篇關(guān)于pytorch tensorboard可視化的使用詳解的文章就介紹到這了,更多相關(guān)pytorch tensorboard內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python的scrapy框架之Pipeline文件的用法詳解
這篇文章主要介紹了python的scrapy框架之Pipeline文件的用法詳解,Pipeline是一個(gè)獨(dú)立的模塊,用于處理從Spider中提取的Item對(duì)象,實(shí)現(xiàn)對(duì)數(shù)據(jù)的進(jìn)一步處理、存儲(chǔ)和清洗等操作,下面將詳細(xì)介紹Scrapy框架中Pipeline的用法,需要的朋友可以參考下2023-10-10
分布式全文檢索引擎ElasticSearch原理及使用實(shí)例
這篇文章主要介紹了分布式全文檢索引擎ElasticSearch原理及使用實(shí)例,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-11-11
python3中@dataclass的實(shí)現(xiàn)示例
@dataclass?是 Python 3.7 引入的一個(gè)裝飾器,用于方便地定義符合數(shù)據(jù)類協(xié)議的類,本文主要介紹了python3中@dataclass的實(shí)現(xiàn)示例,感興趣的可以了解一下2024-02-02
python 使用elasticsearch 實(shí)現(xiàn)翻頁的三種方式
這篇文章主要介紹了python 使用elasticsearch 實(shí)現(xiàn)翻頁的三種方式,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07
django manage.py擴(kuò)展自定義命令方法
今天小編就為大家分享一篇django manage.py擴(kuò)展自定義命令方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-05-05

