Поделиться через


PyTorchConfiguration Класс

Представляет сведения о конфигурации для распределенных заданий PyTorch.

Этот класс используется в RunConfiguration классе.

Конструктор Класса 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