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

communication_backend
str
varsayılan değer: Nccl

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.

process_count
int
varsayılan değer: None

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

node_count
int
varsayılan değer: 1

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