請(qǐng)人做網(wǎng)站收費(fèi)多少錢杭州谷歌推廣
文章目錄
- 1 wandb介紹
- 2 wandb安裝
- 2.1 注冊(cè)wandb賬號(hào)
- 2.2 創(chuàng)建項(xiàng)目并獲得密鑰
- 2.3 安裝wandb并登錄
- 3 wandb本地部署
- 3.1 設(shè)置wandb運(yùn)行模式
- 3.2 云端查看運(yùn)行數(shù)據(jù)
- 4 總結(jié)
1 wandb介紹
Wandb(Weights & Biases)是一個(gè)用于跟蹤、可視化和協(xié)作機(jī)器學(xué)習(xí)實(shí)驗(yàn)的開(kāi)源工具和平臺(tái)。它為機(jī)器學(xué)習(xí)項(xiàng)目提供了一種簡(jiǎn)單而強(qiáng)大的方式來(lái)記錄實(shí)驗(yàn)參數(shù)、跟蹤指標(biāo)、可視化模型和數(shù)據(jù),并與團(tuán)隊(duì)成員共享結(jié)果。
2 wandb安裝
2.1 注冊(cè)wandb賬號(hào)
在使用wandb庫(kù)時(shí)需要在它的官網(wǎng)注冊(cè)一個(gè)賬號(hào),網(wǎng)址是:https://wandb.ai/site。
’
2.2 創(chuàng)建項(xiàng)目并獲得密鑰
成功注冊(cè)并登錄后在界面的左上角選擇Create new project
進(jìn)行新建項(xiàng)目,下面是新建項(xiàng)目頁(yè)面:

填寫(xiě)好項(xiàng)目名稱和描述后便可以得到該項(xiàng)目的密鑰API Key:

2.3 安裝wandb并登錄
按照安裝界面上的提示使用pip命令安裝wandb包:
pip install wandb
在安裝好后跳轉(zhuǎn)到你要使用wandb的程序所在的命令行中,使用下面的命令進(jìn)行登錄并按提示輸入API Key 信息:
wandb login
具體的登錄方式以及wandb在程序中的相關(guān)配置方法可以參考下面這篇博客:深度學(xué)習(xí)之wandb的基本使用
3 wandb本地部署
wandb最為人稱贊的地方在于它能夠以多種方式可視化神經(jīng)網(wǎng)絡(luò)、實(shí)時(shí)監(jiān)督訓(xùn)練過(guò)程,并實(shí)現(xiàn)多人協(xié)同操作。但是實(shí)現(xiàn)這些功能的基礎(chǔ)是需要聯(lián)網(wǎng)。在一些實(shí)際使用中,我們可能只想在本地訓(xùn)練,在訓(xùn)練完畢后統(tǒng)一可視化查看,這時(shí)就要求wandb能夠在本地使用,并保存數(shù)據(jù)以待我們需要的時(shí)候進(jìn)行聯(lián)網(wǎng)查看。
因此,wandb內(nèi)部集成了一套本地部署方法, 具體使用方法為:
3.1 設(shè)置wandb運(yùn)行模式
為了本地使用wandb,需要將其運(yùn)行方式設(shè)置為本地(即“dryrun
”)模式。相應(yīng)的代碼演示:
import wandb
# 初始化wandb
run = wandb.init(mode="dryrun")
# 其他訓(xùn)練代碼
...
在這種模式下,wandb 會(huì)把所有的日志數(shù)據(jù)保存在本地磁盤,而不是上傳到云端。
3.2 云端查看運(yùn)行數(shù)據(jù)
當(dāng)需要查看網(wǎng)絡(luò)的運(yùn)行數(shù)據(jù)時(shí),重新接入網(wǎng)絡(luò)并使用 wandb sync
命令便可同步本地運(yùn)行數(shù)據(jù)到 wandb 云服務(wù)。以下是相應(yīng)的命令:
wandb sync /path/to/dryrun/directory
請(qǐng)注意,這里的“/path/to/dryrun/directory”表示的是當(dāng)前本地 dryrun 目錄的路徑,在具體使用時(shí)應(yīng)該根據(jù)當(dāng)前項(xiàng)目路徑進(jìn)行具體替換。
這樣,即使在沒(méi)有網(wǎng)絡(luò)連接的情況下,也可以跟蹤機(jī)器學(xué)習(xí)實(shí)驗(yàn),并在恢復(fù)網(wǎng)絡(luò)連接后上傳實(shí)驗(yàn)數(shù)據(jù)。
4 總結(jié)
在一些大型的深度學(xué)習(xí)項(xiàng)目中,合理的應(yīng)用wandb能夠幫助我們快速分析網(wǎng)絡(luò)架構(gòu)和模型組成,便于高效的開(kāi)展針對(duì)性的改進(jìn)工作。本文僅總結(jié)了wandb的安裝方法和本地部署方法,如果讀者有需要可以去wandb官網(wǎng)的使用文檔上學(xué)習(xí)更多的使用方法:https://docs.wandb.ai/