PyTorchConfiguration Classe

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

Nome Descrizione
communication_backend
str

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

Valore predefinito: Nccl
process_count
int

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

Valore predefinito: None
node_count
int

Numero di nodi da usare per il processo.

Valore predefinito: 1