PyTorchConfiguration Kelas

  • Referensi

Mewakili informasi konfigurasi untuk pekerjaan PyTorch terdistribusi.

Kelas ini digunakan dalam kelas RunConfiguration.

Konstruktor Class PyTorchConfiguration.

Warisan
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Konstruktor

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

Parameter

communication_backend
str
nilai default: Nccl

Backend komunikasi untuk digunakan untuk pelatihan terdistribusi PyTorch. Backend yang didukung adalah "Nccl" dan "Gloo". Defaultnya adalah "Nccl".

process_count
int
nilai default: None

Jumlah proses per node yang diluncurkan untuk pekerjaan. Secara default nilai akan diatur ke node_count.

node_count
int
nilai default: 1

Jumlah node yang digunakan untuk pekerjaan.