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

communication_backend

str

valor predefinido: Nccl

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".

process_count

int

valor predefinido: None

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

node_count

int

valor predefinido: 1

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