본문으로 건너뛰기

스윕 초기화

W&B는 클라우드(표준) 또는 로컬(지역)에서 하나 이상의 기계를 통해 스윕을 관리하기 위한 스윕 컨트롤러를 사용합니다. run이 완료된 후, 스윕 컨트롤러는 실행할 새로운 run을 설명하는 새로운 지침 세트를 발행합니다. 이러한 지침은 실제로 run을 수행하는 에이전트에 의해 수집됩니다. 전형적인 W&B 스윕에서 컨트롤러는 W&B 서버에 있습니다. 에이전트는 당신의 기계(들)에 있습니다.

다음 코드조각은 CLI 및 Jupyter 노트북 또는 파이썬 스크립트 내에서 스윕을 초기화하는 방법을 보여줍니다.

주의
  1. 스윕을 초기화하기 전에, YAML 파일이나 스크립트 내에서 중첩된 파이썬 딕셔너리 오브젝트로 스윕 구성이 정의되어 있는지 확인하세요. 자세한 내용은 스윕 구성 정의를 참조하세요.
  2. W&B 스윕과 W&B Run은 같은 프로젝트에 있어야 합니다. 따라서, W&B를 초기화할 때 제공하는 이름(wandb.init)은 W&B 스윕을 초기화할 때 제공하는 프로젝트 이름(wandb.sweep)과 일치해야 합니다.

W&B SDK를 사용하여 스윕을 초기화하세요. 스윕 구성 딕셔너리를 sweep 파라미터에 전달하세요. 선택적으로 출력물을 저장하고 싶은 프로젝트의 이름을 프로젝트 파라미터(project)에 제공하세요. 프로젝트가 지정되지 않은 경우, run은 "분류되지 않은" 프로젝트에 배치됩니다.

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에는 엔티티 이름과 프로젝트 이름이 포함됩니다. 스윕 ID를 기록해두세요.

Was this page helpful?👍👎