setup() - Weights & Biases Documentation

関数 wandb.setup()

現在のプロセスおよびその子プロセスで W&B を使用できるように準備します。通常、これは wandb.init() から暗黙的に呼び出されるため、たいていは意識する必要はありません。 wandb を複数のプロセスで使用する場合は、子プロセスを開始する前に親プロセスで wandb.setup() を呼び出すことで、パフォーマンスやリソース使用率が向上することがあります。 wandb.setup() は os.environ を変更するため、子プロセスが変更後の環境変数を継承することが重要です。 wandb.teardown() も参照してください。

settings: 'Settings | None' = None

引数

Settings | None

グローバルに適用する設定です。これらは後続の wandb.init() 呼び出しで上書きできます。

例

import multiprocessing

import wandb


def run_experiment(params):
    with wandb.init(config=params):
        # 実験を実行する
        pass


if __name__ == "__main__":
    # バックエンドを起動してグローバル設定を行う
    wandb.setup(settings={"project": "my_project"})

    # 実験パラメーターを定義する
    experiment_params = [
        {"learning_rate": 0.01, "epochs": 10},
        {"learning_rate": 0.001, "epochs": 20},
    ]

    # 複数のプロセスを起動し、それぞれ個別の実験を実行する
    processes = []
    for params in experiment_params:
        p = multiprocessing.Process(target=run_experiment, args=(params,))
        p.start()
        processes.append(p)

    # すべてのプロセスの完了を待つ
    for p in processes:
        p.join()

    # オプション: バックエンドを明示的にシャットダウンする
    wandb.teardown()