sweep agent を開始する

1つ以上のマシンで W&B Sweep Agent を開始または停止します。

1 つまたは複数のマシン上の、1 つまたは複数のエージェントで Sweeps を開始します。 sweep agent は、 sweep を初期化 した際に定義された sweep configuration を使用して、異なる ハイパーパラメーター の組み合わせを探索します。 W&B は、 sweep agent が試行する ハイパーパラメーター の組み合わせごとに、新しい Run を作成します。

sweep の一時停止、再開、停止、またはキャンセル方法については、 Sweeps の管理 を参照してください。

続行する前に、以下を確認してください：

• トレーニングスクリプト が、 W&B で ハイパーパラメーター の組み合わせを作成および追跡するように 設定 されていること。詳細については、 コードに W&B を追加する を参照してください。
• sweep 用の 設定ファイル が定義されていること。

以下の コードスニペット は、 CLI 、および Jupyter Notebook または Python スクリプト 内で エージェント を開始する方法を示しています。どちらの方法でも、 sweep を初期化したときに W&B から返された sweep ID を指定します。 sweep ID の形式は次のとおりです。

entity/project/sweep_ID

ここで：

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

CLI

sweep を開始するには、 wandb agent コマンド を使用します。 sweep を初期化したときに W&B から返された sweep ID を指定してください。

以下の コードスニペット をコピーして貼り付け、 sweep_id を実際の sweep ID に置き換えてください。

wandb agent sweep_id

Python script or notebook

sweep を開始するには、 wandb.agent() を使用します。 sweep を初期化したときに W&B から返された sweep ID と、 トレーニングスクリプト のエントリポイントとなる関数の名前を指定します。

以下の コードスニペット をコピーして貼り付け、 <sweep_id> を実際の sweep ID に、 <function_name> をトレーニング関数の名前に置き換えてください。

wandb.agent(sweep_id="<sweep_id>", function="<function_name>")

この方法を使用する場合の トレーニングスクリプト のセットアップ例については、「コードに W&B を追加する」の Python script or notebook タブ を参照してください。

マルチプロセッシング

Python 標準ライブラリの multiprocessing または PyTorch の pytorch.multiprocessing パッケージを使用する場合、 wandb.agent() と wandb.sweep() の呼び出しを if __name__ == '__main__': で囲む必要があります。例：

if __name__ == '__main__':
    wandb.agent(sweep_id="<sweep_id>", function="<function>", count="<count>")

この慣習に従ってコードを囲むことで、スクリプトが直接実行されたときのみ実行され、ワーカープロセスでモジュールとしてインポートされたときには実行されないことが保証されます。

マルチプロセッシングの詳細については、 Python 標準ライブラリ multiprocessing または PyTorch multiprocessing を参照してください。 if __name__ == '__main__': の慣習については、 https://realpython.com/if-name-main-python/ を参照してください。

​

sweep agent が試行する Run の数を制限する

ランダム探索および ベイズ探索 は永続的に実行されます。 コマンドライン 、 Python スクリプト 内、または Sweeps UI から プロセス を停止する必要があります。

sweep agent が試行すべき Run の回数を指定します。以下の コードスニペット は、 CLI 、 Jupyter Notebook 、および Python スクリプト 内で、 W&B Runs の最大数を設定する方法を示しています。

CLI

まず、 wandb sweep コマンド で sweep を初期化します。詳細については、 Sweeps の初期化 を参照してください。

wandb sweep config.yaml

次に、 count フラグに整数値を渡して、試行する Run の最大数を設定します。

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

Python script or notebook

まず、 sweep を初期化します。詳細については、 Sweeps の初期化 を参照してください。

sweep_id = wandb.sweep(sweep_config)

次に、 sweep ジョブを開始します。 sweep の初期化で生成された sweep ID を指定します。 count パラメータ に整数値を渡して、試行する Run の最大数を設定します。

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

sweep agent が終了した後に、同じスクリプトまたは ノートブック 内で新しい Run を開始する場合は、新しい Run を開始する前に wandb.teardown() を呼び出す必要があります。