Sdílet prostřednictvím

PyTorchConfiguration Třída

  • Reference

Představuje informace o konfiguraci pro distribuované úlohy PyTorch.

Tato třída se používá ve RunConfiguration třídě .

Konstruktor třídy PyTorchConfiguration.

Dědičnost
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Konstruktor

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

Parametry

Name Description
communication_backend
str

Komunikační back-end, který se má použít pro distribuované trénování PyTorchu. Podporované back-endy jsou Nccl a Gloo. Výchozí hodnota je Nccl.

Default value: Nccl
process_count
int

Celkový počet procesů, které se mají spustit pro úlohu. Ve výchozím nastavení se hodnota nastaví na node_count.

Default value: None
node_count
int

Počet uzlů, které se mají použít pro úlohu.

Default value: 1