PyTorchConfiguration Osztály

  • Referencia

Az elosztott PyTorch-feladatok konfigurációs adatait jelöli.

Ezt az osztályt használja a RunConfiguration osztály.

PyTorchConfiguration konstruktor osztály.

Öröklődés
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Konstruktor

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

Paraméterek

communication_backend
str
alapértelmezett érték: Nccl

A PyTorch elosztott betanításához használható kommunikációs háttérrendszer. A támogatott háttérrendszer a "Nccl" és a "Gloo". Az alapértelmezett érték az "Nccl".

process_count
int
alapértelmezett érték: None

A feladathoz elindítandó folyamatok teljes száma. Az érték alapértelmezés szerint a node_countértékre lesz állítva.

node_count
int
alapértelmezett érték: 1

A feladathoz használandó csomópontok száma.