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

学習設定でコマンドライン引数を受け取るようにしていれば、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}