Compartir a través de

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

Nombre Description
communication_backend
str

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".

Valor predeterminado: Nccl
process_count
int

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

Valor predeterminado: None
node_count
int

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

Valor predeterminado: 1