sweep エージェントを開始または停止する

W&B Sweep を 1 台以上のマシン上の 1 台以上のエージェントで開始します。W&B Sweep エージェントは、ハイパーパラメーターを取得するために W&B Sweep を初期化したときにローンンチされた W&B サーバーにクエリを送り、それらを使用してモデル トレーニングを実行します。

W&B Sweep エージェントを開始するには、W&B Sweep を初期化したときに返された W&B Sweep ID を指定します。W&B Sweep ID の形式は次のとおりです。

entity/project/sweep_ID

ここで:

• entity: W&B のユーザー名またはチーム名。
• project: W&B Run の出力を保存したいプロジェクトの名前。プロジェクトが指定されていない場合、run は「未分類」プロジェクトに置かれます。
• sweep_ID: W&B によって生成される疑似ランダムな一意の ID。

Jupyter ノートブックまたは Python スクリプト内で W&B Sweep エージェントを開始する場合、W&B Sweep が実行する関数の名前を指定します。

次のコードスニペットは、W&B でエージェントを開始する方法を示しています。ここでは、既に設定ファイルを持っており、W&B Sweep を初期化済みであると仮定しています。設定ファイルを定義する方法について詳しくは、Define sweep configuration を参照してください。

• CLI
• Python script or notebook

wandb agent sweep_id

wandb.agent(sweep_id=sweep_id, function=function_name)

W&B エージェントを停止

オプションで、Sweep agent が試みるべき W&B Runs の数を指定します。以下のコードスニペットは、Jupyter ノートブック、Python スクリプト内で最大の W&B Runs 数を設定する方法を示しています。

• Python script or notebook
• CLI

sweep_id = wandb.sweep(sweep_config)

sweep_id, count = "dtzl1o7u", 10
wandb.agent(sweep_id, count=count)

wandb sweep config.yaml

NUM=10
SWEEPID="dtzl1o7u"
wandb agent --count $NUM $SWEEPID