PyTorchConfiguration Класс
Представляет сведения о конфигурации для распределенных заданий PyTorch.
Этот класс используется в RunConfiguration классе.
Конструктор Класса PyTorchConfiguration.
Конструктор
PyTorchConfiguration(communication_backend='Nccl', process_count=None, node_count=1)
Параметры
Имя | Описание |
---|---|
communication_backend
|
Серверная часть обмена данными, используемая для распределенного обучения PyTorch. Поддерживаемые серверные части : "Nccl" и "Gloo". Значение по умолчанию — Nccl. Default value: Nccl
|
process_count
|
Общее количество процессов для запуска задания. По умолчанию для значения будет задано Default value: None
|
node_count
|
Число узлов, используемых для задания. Default value: 1
|