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