setup() - CoreWeave Docs

함수 wandb.setup()

현재 프로세스와 자식 프로세스에서 사용할 수 있도록 W&B를 준비합니다. wandb.init()에서 암시적으로 호출되므로 일반적으로 신경 쓰지 않아도 됩니다. 여러 프로세스에서 wandb를 사용할 때는 자식 프로세스를 시작하기 전에 부모 프로세스에서 wandb.setup()을 호출하면 성능과 리소스 사용량을 개선할 수 있습니다. wandb.setup()은 os.environ을 수정하므로, 자식 프로세스가 수정된 환경 변수를 상속하도록 해야 합니다. wandb.teardown()도 참고하세요.

settings: 'Settings | None' = None

인수

Settings | None

전역으로 적용할 설정입니다. 이후 wandb.init() Call로 재정의할 수 있습니다.

예시

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

​함수 wandb.setup()

​인수

​예시

함수 wandb.setup()

인수

예시