Sweeps 에서 커스텀 CLI 코맨드를 어떻게 사용하나요?

트레이닝 설정이 커맨드라인 인수를 전달하는 경우, 커스텀 CLI 코맨드와 함께 W&B Sweeps 를 사용할 수 있습니다.

아래 예시의 코드조각은 사용자가 train.py라는 이름의 Python 스크립트를 트레이닝하며 스크립트가 파싱할 값들을 제공하는 bash 터미널을 보여줍니다:

/usr/bin/env python train.py -b \
    your-training-config \
    --batchsize 8 \
    --lr 0.00001

커스텀 코맨드를 구현하려면 YAML 파일의 command 키를 수정하십시오. 이전 예시를 바탕으로 한 구성은 다음과 같습니다:

program:
  train.py
method: grid
parameters:
  batch_size:
    value: 8
  lr:
    value: 0.0001
command:
  - ${env}
  - python
  - ${program}
  - "-b"
  - your-training-config
  - ${args}

${args} 키는 sweep 구성의 모든 파라미터로 확장되며, argparse 형식에 맞춰 --param1 value1 --param2 value2 형태로 포맷팅됩니다.

argparse 이외의 추가 인수를 처리하려면 다음과 같이 구현하십시오:

# parser 설정
parser = argparse.ArgumentParser()
args, unknown = parser.parse_known_args()

환경에 따라 python이 Python 2를 가리킬 수 있습니다. Python 3가 호출되도록 보장하려면 코맨드 설정에서 python3를 사용하십시오:

program:
  script.py
command:
  - ${env}
  - python3
  - ${program}
  - ${args}