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