PyTorchConfiguration Classe

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

communication_backend

str

valor padrão: Nccl

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

process_count

int

valor padrão: None

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

node_count

int

valor padrão: 1

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