PyTorchConfiguration Classe

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

Essa classe é usada na RunConfiguration classe.

Construtor de classe 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 O 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