🔥

Wandb offline mode

2023/05/10に公開

W&Bにはoffline modeがある。
これを使うとlocalに保存して、後でweb dashboardに結果をupload することができる。

import wandb
import random

mode = 'offline'

wandb.init(project="wandb_test",
           anonymous="allow",
		   mode=mode)

for step in range(10):
    wandb.log({
          "acc": random.random(),
          "loss": random.random()
      }, step=step)

wandb.finish()

updalodには

wandb sync --sync-all

実験結果はcurrent dirのwandbというdirに保存される
以下はonlineとofflineを1回ずつ実験した結果
offlineで実行するとoffline-runとなる

➜  wandb_test ls wandb
offline-run-20230510_113548-s5vjrt42
run-20230510_113500-rdyfflkg

offlineの結果がsyncされてるかどうかは、offline-runの下に
run-s5vjrt42.wandb.synced
が作成されているかで判断しているようだ。

学習途中のofflineのlogをsyncすることはできない。
onlineのように途中経過をofflineで確認する方法はなさそう😅

保存先がlocalhostになってしまった時、cloudに戻すには以下を実行する

wandb login --cloud

Discussion