PyTorchConfiguration クラス

  • リファレンス

分散 PyTorch ジョブの構成情報を表します。

このクラスは RunConfiguration クラスで使用されます。

PyTorchConfiguration クラス コンストラクター。

継承
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

コンストラクター

PyTorchConfiguration(communication_backend='Nccl', process_count=None, node_count=1)

パラメーター

communication_backend
str
既定値: Nccl

PyTorch 分散トレーニングに使用する通信バックエンド。 サポートされているバックエンドは "Nccl" と "Gloo" です。 既定値は "Nccl" です。

process_count
int
既定値: None

ジョブで起動するプロセスの合計数です。 既定では、値は node_count に設定されます。

node_count
int
既定値: 1

ジョブに使用するノードの数。