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

要用於作業的節點數目。