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

Число узлов, используемых для задания.