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.

Konstruktor

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

Parametry

Name	Description
communication_backend	str Komunikace back-endu pro distribuované trénování PyTorch. 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í pro úlohu spustit. Ve výchozím nastavení bude hodnota nastavena node_countna hodnotu . Default value: None
node_count	int Počet uzlů, které se mají pro úlohu použít. Default value: 1