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

W&B スイープは、トレーニング設定がコマンドライン引数を通過する場合、カスタム 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}