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

communication_backend
str
standardvärde: Nccl

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

process_count
int
standardvärde: None

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

node_count
int
standardvärde: 1

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