다음을 통해 공유

PyTorchConfiguration 클래스

  • 참조

분산 PyTorch 작업에 대한 구성 정보를 나타냅니다.

이 클래스에서 사용 되는 RunConfiguration 클래스입니다.

클래스 PyTorchConfiguration 생성자입니다.

상속
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

생성자

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

매개 변수

Name Description
communication_backend
str

PyTorch 분산 학습에 사용할 통신 백 엔드입니다. 지원되는 백 엔드는 “Nccl” 및 “Gloo”입니다. 기본값은 “Nccl”입니다.

Default value: Nccl
process_count
int

작업에 대해 시작할 총 프로세스 수입니다. 기본적으로 이 값은 node_count로 설정됩니다.

Default value: None
node_count
int

작업에 사용할 노드 수입니다.

Default value: 1