PyTorchConfiguration Class

Represents configuration information for distributed PyTorch jobs.

This class is used in the RunConfiguration class.

Inheritance
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Constructor

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

Parameters

communication_backend
str

Communication backend to use for PyTorch distributed training. Supported backends are "Nccl" and "Gloo". Default is "Nccl".

node_count
int

The number of nodes to use for the job.