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

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

規定値: Nccl
process_count
int

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

規定値: None
node_count
int

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

規定値: 1