스윕에서 커스텀 CLI 명령을 어떻게 사용하나요?

트레이닝 설정이 명령줄 인자를 받도록 되어 있다면, W&B Sweeps를 커스텀 CLI 명령과 함께 사용할 수 있습니다.

아래 예시에서 코드 스니펫은 bash 터미널에서 train.py라는 Python 스크립트를 사용해 트레이닝을 수행하면서, 스크립트가 파싱할 값을 인자로 전달하는 모습을 보여줍니다:

/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} 키는 스윕 설정에 포함된 모든 파라미터를 argparse용 형식인 --param1 value1 --param2 value2로 확장합니다.

argparse로 전달되지 않는 추가 인수를 사용하려면 다음을 구현하세요:

parser = argparse.ArgumentParser()
args, unknown = parser.parse_known_args()

환경에 따라 python이 Python 2를 가리킬 수 있습니다. 반드시 Python 3가 실행되도록 하려면 명령 설정에서 python3를 사용하세요:

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