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/