setup()

GitHub source

​

function setup

setup(settings: 'Settings | None' = None) → _WandbSetup

현재 프로세스와 그 자식 프로세스에서 사용할 수 있도록 W&B를 준비합니다.

일반적으로 wandb.init()에 의해 암시적으로 호출되므로 이 함수를 직접 호출하지 않아도 무방합니다.

멀티 프로세스 환경에서 wandb를 사용하는 경우, 자식 프로세스를 시작하기 전에 부모 프로세스에서 wandb.setup()을 호출하면 성능과 리소스 활용도를 높일 수 있습니다.

wandb.setup()은 os.environ을 수정하며, 자식 프로세스가 수정된 환경 변수를 상속받는 것이 중요합니다.

wandb.teardown()도 함께 참고하세요.

Args:

• settings: 전역적으로 적용할 설정입니다. 이후 호출되는 wandb.init()에 의해 이 설정이 덮어씌워질 수 있습니다.

Example:

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