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()