Dela via

PyTorchConfiguration Klass

Representerar konfigurationsinformation för distribuerade PyTorch-jobb.

Den här klassen används i RunConfiguration klassen.

Konstruktor för klass PyTorchConfiguration.

Konstruktor

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

Parametrar

Name Description
communication_backend
str

Kommunikationsserverdel som ska användas för PyTorch-distribuerad utbildning. Serverdelar som stöds är "Nccl" och "Gloo". Standardvärdet är "Nccl".

Standardvärde: Nccl
process_count
int

Det totala antalet processer som ska startas för jobbet. Som standard anges värdet till node_count.

Standardvärde: None
node_count
int

Antalet noder som ska användas för jobbet.

Standardvärde: 1