Condividi tramite

PyTorchConfiguration Classe

  • Riferimento

Rappresenta le informazioni di configurazione per i processi PyTorch distribuiti.

Questa classe viene utilizzata nella RunConfiguration classe .

Costruttore PyTorchConfiguration della classe.

Ereditarietà
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Costruttore

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

Parametri

communication_backend
str
valore predefinito: Nccl

Back-end di comunicazione da usare per il training distribuito pyTorch. I back-end supportati sono "Nccl" e "Gloo". Il valore predefinito è "Nccl".

process_count
int
valore predefinito: None

Numero totale di processi da avviare per il processo. Per impostazione predefinita, il valore verrà impostato su node_count.

node_count
int
valore predefinito: 1

Numero di nodi da usare per il processo.