Initialize sweeps

W&B は Sweep Controller を使用してクラウド（標準）またはローカル（ローカル）で複数のマシンにわたる sweeps を管理します。run が完了すると、sweep controller は新しい run を実行するための新しい指示セットを発行します。これらの指示は、実際に run を実行する agents に拾われます。通常の W&B Sweep では、controller は W&B サーバー上にあり、agents は あなた のマシン上に存在します。

以下のコードスニペットは、CLI および Jupyter Notebook や Python スクリプト内で sweeps を初期化する方法を示しています。

注意

1. Sweep を初期化する前に、YAML ファイルまたはスクリプト内のネストされた Python 辞書オブジェクトで定義された sweep configuration があることを確認してください。詳細については、Define sweep configuration を参照してください。
2. W&B Sweep と W&B Run は同じプロジェクト内になければなりません。したがって、W&B を初期化するときに提供する名前 (wandb.init) は、W&B Sweep を初期化するときに提供するプロジェクトの名前 (wandb.sweep) と一致する必要があります。

Python script or Jupyter Notebook

W&B SDK を使用して sweep を初期化します。sweep configuration 辞書を sweep パラメータに渡します。また、W&B Run の出力を保存したいプロジェクトの名前を project パラメータ (project) にオプションで提供できます。プロジェクトが指定されていない場合、run は「Uncategorized」というプロジェクトに入れられます。

import wandb

# 実例 sweep configuration
sweep_configuration = {
    "method": "random",
    "name": "sweep",
    "metric": {"goal": "maximize", "name": "val_acc"},
    "parameters": {
        "batch_size": {"values": [16, 32, 64]},
        "epochs": {"values": [5, 10, 15]},
        "lr": {"max": 0.1, "min": 0.0001},
    },
}

sweep_id = wandb.sweep(sweep=sweep_configuration, project="project-name")

wandb.sweep 関数は sweep ID を返します。sweep ID にはエンティティ名とプロジェクト名が含まれます。sweep ID をメモしておいてください。

CLI

W&B CLI を使用して sweep を初期化します。設定ファイルの名前を指定してください。オプションで、project フラグにプロジェクトの名前を指定できます。プロジェクトが指定されていない場合、W&B Run は「Uncategorized」というプロジェクトに入れられます。

wandb sweep コマンドを使用して sweep を初期化します。次のコード例では、sweeps_demo プロジェクトの sweep を初期化し、config.yaml ファイルを設定に使用します。

wandb sweep --project sweeps_demo config.yaml

このコマンドは sweep ID を出力します。sweep ID にはエンティティ名とプロジェクト名が含まれます。sweep ID をメモしておいてください。