Aracılığıyla paylaş

PyTorchConfiguration Sınıf

  • Başvuru

Dağıtılmış PyTorch işleri için yapılandırma bilgilerini temsil eder.

Bu sınıf sınıfında RunConfiguration kullanılır.

PyTorchConfiguration oluşturucu sınıfı.

Devralma
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Oluşturucu

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

Parametreler

Name Description
communication_backend
str

PyTorch dağıtılmış eğitimi için kullanılacak iletişim arka ucu. Desteklenen arka uçlar "Nccl" ve "Gloo" şeklindedir. Varsayılan değer "Nccl"dir.

Default value: Nccl
process_count
int

İş için başlatacak toplam işlem sayısı. Varsayılan olarak değer olarak node_countayarlanır.

Default value: None
node_count
int

İş için kullanılacak düğüm sayısı.

Default value: 1