PyTorchConfiguration Třída
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._AbstractRunConfigElementPyTorchConfiguration
Konstruktor
PyTorchConfiguration(communication_backend='Nccl', process_count=None, node_count=1)
Parametry
- communication_backend
- str
výchozí hodnota: Nccl
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.
- process_count
- int
výchozí hodnota: None
Celkový počet procesů, které se mají spustit pro úlohu. Ve výchozím nastavení se hodnota nastaví na node_count
.
Váš názor
https://aka.ms/ContentUserFeedback.
Připravujeme: V průběhu roku 2024 budeme postupně vyřazovat problémy z GitHub coby mechanismus zpětné vazby pro obsah a nahrazovat ho novým systémem zpětné vazby. Další informace naleznete v tématu:Odeslat a zobrazit názory pro