カスタム CLI コマンドを Sweeps で使用するにはどうすればよいですか？

トレーニング設定がコマンドライン引数を受け取る場合、カスタム 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 configuration 内のすべてのパラメータに展開され、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}