Udostępnij za pośrednictwem

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

Nazwa Opis
communication_backend
str

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

Domyślna wartość: Nccl
process_count
int

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

Domyślna wartość: None
node_count
int

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

Domyślna wartość: 1