Share via

PyTorchConfiguration Klasa

  • Odwołanie

Reprezentuje informacje o konfiguracji dla rozproszonych zadań PyTorch.

Ta klasa jest używana w RunConfiguration klasie .

Konstruktor klasy PyTorchConfiguration.

Dziedziczenie
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Konstruktor

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

Parametry

communication_backend
str
wartość domyślna: Nccl

Zaplecze komunikacji do użycia na potrzeby trenowania rozproszonego PyTorch. Obsługiwane zaplecza to "Nccl" i "Gloo". Wartość domyślna to "Nccl".

process_count
int
wartość domyślna: None

Całkowita liczba procesów do uruchomienia dla zadania. Domyślnie wartość zostanie ustawiona na node_countwartość .

node_count
int
wartość domyślna: 1

Liczba węzłów do użycia dla zadania.