PyTorchConfiguration 클래스
분산 PyTorch 작업에 대한 구성 정보를 나타냅니다.
이 클래스는 클래스에서 RunConfiguration 사용됩니다.
클래스 PyTorchConfiguration 생성자입니다.
생성자
PyTorchConfiguration(communication_backend='Nccl', process_count=None, node_count=1)매개 변수
| Name | Description |
|---|---|
|
communication_backend
|
PyTorch 분산 학습에 사용할 통신 백 엔드입니다. 지원되는 백 엔드는 "Nccl" 및 "Gloo"입니다. 기본값은 "Nccl"입니다. Default value: Nccl
|
|
process_count
|
작업에 대해 시작할 총 프로세스 수입니다. 기본적으로 값은 .로 Default value: None
|
|
node_count
|
작업에 사용할 노드 수입니다. Default value: 1
|