PyTorchConfiguration Klasse

  • Referenz

Stellt Konfigurationsinformationen für verteilte PyTorch-Aufträge dar.

Diese Klasse wird in der RunConfiguration-Klasse verwendet.

PyTorchConfiguration-Konstruktor der Klasse.

Vererbung
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
Standardwert: Nccl

Kommunikations-Back-End für verteiltes PyTorch-Training. Unterstützte Back-Ends sind „Nccl“ und „Gloo“. Der Standardwert ist „Nccl“.

process_count
int
Standardwert: None

Die Gesamtanzahl von Prozessen, die für den Auftrag gestartet werden sollen. Der Wert wird standardmäßig auf die Knotenanzahl (node_count) festgelegt.

node_count
int
Standardwert: 1

Die Anzahl von Knoten, die für den Auftrag verwendet werden sollen.