PyTorchConfiguration Classe

  • Referência

Representa informações de configuração para trabalhos PyTorch distribuídos.

Essa classe é usada na classe RunConfiguration.

Construtor de classe PyTorchConfiguration.

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

Nome Description
communication_backend
str

Back-end de comunicação a ser usado para treinamento distribuído do PyTorch. Os back-ends com suporte são "Nccl" e "Gloo". O padrão é "Nccl".

Valor padrão: Nccl
process_count
int

Número total de processos a serem iniciados para o trabalho. Por padrão, o valor será definido como node_count.

Valor padrão: None
node_count
int

O número de nós que serão usados para o trabalho.

Valor padrão: 1