스윕에서 사용자 정의 CLI 명령을 어떻게 사용하나요?

훈련 설정에서 커맨드라인 인자를 전달한다면, W&B 스윕을 사용자 정의 CLI 명령과 함께 사용할 수 있습니다.

아래 예시 코드 스니펫은 사용자가 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} 키는 스윕 설정의 모든 파라미터를 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}