Compartir a través de

PyTorchConfiguration Clase

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

Esta clase se usa en la RunConfiguration clase .

Constructor PyTorchConfiguration de clase.

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 va a 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 van a iniciar 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