Partilhar via

PyTorchConfiguration Classe

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

Esta classe é usada na RunConfiguration classe.

Classe PyTorchConfiguration construtor.

Construtor

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

Parâmetros

Name Description
communication_backend
str

Backend de comunicação para uso no treinamento distribuído do PyTorch. Os backends suportados são "Nccl" e "Gloo". O padrão é "Nccl".

Default value: Nccl
process_count
int

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

Default value: None
node_count
int

O número de nós a serem usados para o trabalho.

Default value: 1