W&B にメトリクスをオフラインで保存し、後で同期することはできますか?

wandb.init はデフォルトでメトリクスをリアルタイムでクラウドに同期するプロセスを開始します。オフラインで使用する場合は、オフラインモードを有効にし、後で同期できるように2つの環境変数を設定してください。

次の環境変数を設定します:

  1. WANDB_API_KEY=$KEY、ここで $KEY はあなたの settings page から取得した API キーです。
  2. WANDB_MODE="offline"

スクリプトでこれを実装する例を以下に示します:

import wandb
import os

os.environ["WANDB_API_KEY"] = "YOUR_KEY_HERE"  # あなたの API キーをここに
os.environ["WANDB_MODE"] = "offline"  # オフラインモードを設定

config = {
    "dataset": "CIFAR10",  # データセットを指定
    "machine": "offline cluster",  # オフライン クラスターを指定
    "model": "CNN",  # モデルを指定
    "learning_rate": 0.01,  # 学習率を指定
    "batch_size": 128,  # バッチサイズを指定
}

wandb.init(project="offline-demo")  # W&B プロジェクトを初期化

for i in range(100):
    wandb.log({"accuracy": i})  # メトリクスをログ

サンプルのターミナル出力は以下の通りです:

作業が完了した後、データをクラウドに同期するために以下のコマンドを実行します:

wandb sync wandb/dryrun-folder-name