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

トレーニングの設定でコマンドライン引数を渡すようにしていれば、W&B Sweeps でカスタム 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}

The ${args} キーは、sweep 設定内のすべてのパラメーターに展開され、argparse 用に --param1 value1 --param2 value2 の形式になります。

argparse 以外の追加の引数については、以下を実装します。

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

環境によっては、python が Python 2 を指している場合があります。確実に Python 3 を起動するには、command の設定で python3 を使用してください。

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

---

Sweeps