Dela via

PyTorchConfiguration Klass

  • Referens

Representerar konfigurationsinformation för distribuerade PyTorch-jobb.

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

PyTorchConfiguration-konstruktor för klass.

Arv
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Konstruktor

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

Parametrar

Name Description
communication_backend
str

Kommunikationsserverdelen som ska användas för PyTorch-distribuerad träning. 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