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