PyTorchConfiguration Classe

  • Référence

Représente les informations de configuration pour les tâches PyTorch distribuées.

Cette classe est utilisée dans la classe RunConfiguration.

Constructeur PyTorchConfiguration de classe.

Héritage
azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement
PyTorchConfiguration

Constructeur

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

Paramètres

communication_backend
str
valeur par défaut: Nccl

Serveur principal de communication à utiliser pour la formation distribuée PyTorch. Les serveurs principaux pris en charge sont « Nccl » et « Gloo ». « Nccl » est utilisé par défaut.

process_count
int
valeur par défaut: None

Nombre total de processus à lancer pour la tâche. Par défaut, la valeur est définie sur node_count.

node_count
int
valeur par défaut: 1

Nombre de nœuds à dédier au travail.