PyTorchConfiguration Klasse

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

Name	Beschreibung
communication_backend	str Kommunikations-Back-End für verteiltes PyTorch-Training. Unterstützte Back-Ends sind „Nccl“ und „Gloo“. Der Standardwert ist „Nccl“. Standardwert: Nccl
process_count	int Die Gesamtanzahl von Prozessen, die für den Auftrag gestartet werden sollen. Der Wert wird standardmäßig auf die Knotenanzahl (node_count) festgelegt. Standardwert: None
node_count	int Die Anzahl von Knoten, die für den Auftrag verwendet werden sollen. Standardwert: 1