스윕 에이전트 시작하기

하나 이상의 머신에서 W&B Sweep Agent를 시작하거나 중단합니다.

하나 이상의 머신에서 하나 이상의 에이전트를 실행해 스윕을 시작합니다. 스윕 에이전트는 스윕을 초기화할 때 정의한 스윕 구성(config)을 사용하여 다양한 하이퍼파라미터 조합을 탐색합니다. W&B는 스윕 에이전트가 시도하는 각 하이퍼파라미터 조합마다 새로운 실행을 생성합니다.

스윕을 일시 중지, 재개, 중단 또는 취소하는 방법은 스윕 관리를 참조하세요.

계속하기 전에 다음을 확인하세요:

• 학습 스크립트가 W&B로 하이퍼파라미터 조합을 생성하고 추적하도록 구성되어 있는지 확인합니다. 자세한 내용은 코드에 W&B 추가를 참조하세요.
• 스윕에 대해 정의된 구성 파일을 준비했는지 확인합니다.

다음 코드 스니펫은 CLI와 Jupyter Notebook 또는 Python 스크립트 내에서 에이전트를 시작하는 방법을 보여줍니다. 두 방법 모두, 스윕을 초기화할 때 W&B가 반환한 스윕 ID를 제공해야 합니다. 스윕 ID는 다음 형식을 가집니다:

entity/project/sweep_ID

Where:

• entity: W&B 사용자 이름 또는 팀 이름입니다.
• project: W&B가 실행 결과를 저장할 프로젝트의 이름입니다. 프로젝트를 지정하지 않으면 W&B는 실행을 “Uncategorized”라는 프로젝트에 저장합니다.
• sweep_ID: W&B가 생성하는 의사 난수 기반의 고유 ID입니다.

CLI

wandb agent 명령을 사용해 sweep을 시작합니다. sweep을 초기화할 때 W&B가 반환한 sweep ID를 지정합니다.

아래 코드 스니펫을 복사해 붙여넣고 sweep_id를 자신의 sweep ID로 바꾸십시오:

wandb agent sweep_id

Python 스크립트 또는 노트북

wandb.agent()를 사용해 sweep을 시작합니다. sweep을 초기화할 때 W&B가 반환한 sweep ID와 함께, 학습 스크립트의 엔트리포인트가 되는 함수 이름을 지정합니다.

아래 코드 스니펫을 복사해 붙여넣고 <sweep_id>를 자신의 sweep ID로, <function_name>을 학습 함수 이름으로 바꾸십시오:

wandb.agent(sweep_id="<sweep_id>", function="<function_name>")

이 방법을 사용할 경우 학습 스크립트를 어떻게 설정하는지에 대한 예시는 Add W&B to your code의 Python 스크립트 또는 노트북 탭을 참조하십시오.

Multiprocessing

Python 표준 라이브러리의 multiprocessing 또는 PyTorch의 pytorch.multiprocessing 패키지를 사용하는 경우, wandb.agent()와 wandb.sweep() 호출을 반드시 if __name__ == '__main__':로 감싸야 합니다. 예시는 다음과 같습니다:

if __name__ == '__main__':
    wandb.agent(sweep_id="<sweep_id>", function="<function>", count="<count>")

코드를 이 관용구로 감싸면, 스크립트가 워커 프로세스에서 모듈로 import될 때가 아니라 스크립트를 직접 실행할 때에만 코드가 실행되도록 보장할 수 있습니다.

multiprocessing에 대한 더 자세한 내용은 Python standard library multiprocessing 또는 PyTorch multiprocessing을 참조하십시오. if __name__ == '__main__': 관용구에 대한 정보는 https://realpython.com/if-name-main-python/을 참조하십시오.

​

스윕 에이전트가 시도하는 실행 수 제한하기

Random 및 Bayesian 검색은 종료 조건 없이 계속 실행됩니다. 명령줄, Python 스크립트 내, 또는 Sweeps UI에서 프로세스를 직접 중지해야 합니다.

스윕 에이전트가 시도할 실행 횟수를 지정합니다. 다음 코드 스니펫은 CLI와 Jupyter Notebook, Python 스크립트에서 최대 W&B 실행 수를 설정하는 방법을 보여줍니다.

CLI

먼저 wandb sweep 명령으로 스윕을 초기화합니다. 자세한 내용은 Initialize sweeps를 참조하세요.

wandb sweep config.yaml

다음으로, count 플래그에 정수 값을 전달하여 시도할 최대 실행 수를 설정합니다.

NUM=10
SWEEPID="dtzl1o7u"
wandb agent --count $NUM $SWEEPID

Python script or notebook

먼저 스윕을 초기화합니다. 자세한 내용은 Initialize sweeps를 참조하세요.

sweep_id = wandb.sweep(sweep_config)

다음으로 스윕 작업을 시작합니다. 스윕 시작 시 생성된 스윕 ID를 전달합니다. 시도할 최대 실행 수를 설정하려면 count 매개변수에 정수 값을 전달합니다.

sweep_id, count = "dtzl1o7u", 10
wandb.agent(sweep_id, count=count)

스윕 에이전트가 종료된 이후 동일한 스크립트나 노트북 내에서 새 실행을 시작하려면, 새 실행을 시작하기 전에 wandb.teardown()을 호출해야 합니다.