스윕 초기화 - CoreWeave Docs

스윕 설정을 정의했다면 에이전트를 실행하기 전에 스윕을 초기화하세요. 스윕을 초기화하면 설정이 W&B에 등록되고 스윕 ID가 발급됩니다. 에이전트는 이 스윕 ID로 run 지침을 가져옵니다. 초기화된 스윕은 하나 이상의 머신에서 하이퍼파라미터 검색을 조율할 수 있습니다. W&B는 스윕 컨트롤러를 사용해 하나 이상의 머신에 걸쳐 스윕을 관리합니다. run이 완료되면 컨트롤러가 다음에 실행할 run에 대한 새 지침을 발행하고, 에이전트가 이 지침을 받아 실행합니다. 기본적으로 컨트롤러는 W&B가 호스팅하고, 에이전트는 사용자의 머신에서 실행됩니다. 스윕을 초기화하려면 먼저 YAML 파일이나 스크립트 내 중첩된 Python 딕셔너리 객체로 스윕 설정을 정의해야 합니다. 자세한 내용은 스윕 설정 정의를 참조하세요. 다음 코드 스니펫은 CLI, 그리고 Jupyter 노트북이나 Python 스크립트에서 스윕을 초기화하는 방법을 보여줍니다. 사용 중인 워크플로에 맞는 방법을 선택하세요.

스윕과 run은 같은 프로젝트에 있어야 합니다. W&B를 초기화할 때(wandb.init()) 지정한 프로젝트 이름과 스윕을 초기화할 때(wandb.sweep()) 지정한 프로젝트 이름이 일치해야 합니다.

Python 스크립트 또는 노트북
CLI

W&B SDK를 사용해 스윕을 초기화하세요. 스윕 설정 딕셔너리를 sweep 매개변수에 전달합니다. 필요하면 project 매개변수에 W&B가 run의 출력을 저장할 프로젝트 이름을 지정할 수 있습니다. 프로젝트를 지정하지 않으면 W&B는 run을 “Uncategorized” 프로젝트에 저장합니다.

import wandb

# 스윕 설정 예시
sweep_configuration = {
    "method": "random",
    "name": "sweep",
    "metric": {"goal": "maximize", "name": "val_acc"},
    "parameters": {
        "batch_size": {"values": [16, 32, 64]},
        "epochs": {"values": [5, 10, 15]},
        "lr": {"max": 0.1, "min": 0.0001},
    },
}

sweep_id = wandb.sweep(sweep=sweep_configuration, project="project-name")

wandb.sweep() 함수는 스윕 ID를 반환합니다. 스윕 ID에는 entity 이름과 프로젝트 이름이 포함됩니다. 에이전트를 시작할 때 스윕 ID를 전달해야 하므로 따로 기록해 두세요. 스윕 ID를 확보했다면 하나 이상의 에이전트를 실행해 run을 수행할 수 있습니다.

W&B CLI를 사용해 스윕을 초기화하세요. 설정 파일의 이름을 지정합니다. 필요하면 --project 플래그로 프로젝트 이름을 지정할 수 있습니다. 프로젝트를 지정하지 않으면 W&B는 run을 “Uncategorized” 프로젝트에 저장합니다.wandb sweep 명령으로 스윕을 초기화하세요. 다음 코드 예시는 config.yaml 파일을 설정으로 사용해 sweeps_demo 프로젝트에 스윕을 초기화합니다.

wandb sweep --project sweeps_demo config.yaml

이 명령을 실행하면 스윕 ID가 출력됩니다. 스윕 ID에는 entity 이름과 프로젝트 이름이 포함됩니다. 에이전트를 시작할 때 스윕 ID를 전달해야 하므로 따로 기록해 두세요. 스윕 ID를 확보했다면 하나 이상의 에이전트를 실행해 run을 수행할 수 있습니다.