共用方式為

PyTorchConfiguration 類別

  • 參考

表示分散式 PyTorch 作業的組態資訊。

類別中 RunConfiguration 會使用此類別。

類別 PyTorchConfiguration 建構函式。

繼承
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

建構函式

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

參數

名稱 Description
communication_backend
str

用於 PyTorch 分散式定型的通訊後端。 支援的後端為 「Nccl」 和 「Gloo」。 預設值為 「Nccl」。

預設值: Nccl
process_count
int

要啟動作業的進程總數。 根據預設，值會設定為 node_count 。

預設值: None
node_count
int

要用於作業的節點數目。

預設值: 1