PyTorchConfiguration Klas

  • Referentie

Vertegenwoordigt configuratiegegevens voor gedistribueerde PyTorch-taken.

Deze klasse wordt gebruikt in de RunConfiguration klasse.

Klasse PyTorchConfiguration-constructor.

Overname
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Constructor

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

Parameters

communication_backend
str
standaardwaarde: Nccl

Communicatie-back-end voor gebruik voor gedistribueerde pyTorch-training. Ondersteunde back-ends zijn Nccl en Gloo. De standaardwaarde is Nccl.

process_count
int
standaardwaarde: None

Het totale aantal processen dat voor de taak moet worden gestart. De waarde wordt standaard ingesteld op de node_count.

node_count
int
standaardwaarde: 1

Het aantal knooppunten dat voor de taak moet worden gebruikt.