PyTorchConfiguration 클래스

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

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

클래스 PyTorchConfiguration 생성자입니다.

상속: azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement

PyTorchConfiguration

생성자

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

기본값: Nccl

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

기본값: None

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

기본값: 1

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