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". Default value: Nccl
process_count	int Общее число запускаемых процессов для выполнения задания. По умолчанию будет задано значение node_count. Default value: None
node_count	int Число узлов, используемых для задания. Default value: 1