Partilhar via


PyTorchConfiguration Classe

Representa informações de configuração para tarefas distribuídas do PyTorch.

Esta classe é utilizada na RunConfiguration classe .

Construtor PyTorchConfiguration de Classe.

Herança
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Construtor

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

Parâmetros

Name Description
communication_backend
str

Back-end de comunicação a utilizar para preparação distribuída do PyTorch. Os back-ends suportados são "Nccl" e "Gloo". A predefinição é "Nccl".

Default value: Nccl
process_count
int

O número total de processos a iniciar para a tarefa. Por predefinição, o valor será definido como node_count.

Default value: None
node_count
int

O número de nós a utilizar para a tarefa.

Default value: 1