PyTorchConfiguration Clase

  • Referencia

Representa la información de configuración de los trabajos de PyTorch distribuidos.

Esta clase se usa en la clase RunConfiguration.

Constructor De clase PyTorchConfiguration.

Herencia
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Constructor

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

Parámetros

communication_backend
str
valor predeterminado: Nccl

Back-end de comunicación que se usará para el entrenamiento distribuido de PyTorch. Los back-end admitidos son "Nccl" y "Gloo". El valor predeterminado es "Nccl".

process_count
int
valor predeterminado: None

Número total de procesos que se iniciarán para el trabajo. De forma predeterminada, el valor se establecerá en node_count.

node_count
int
valor predeterminado: 1

El número de nodos que se usarán para el trabajo.