Partager via

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

Nom Description
communication_backend
str

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.

Valeur par défaut: Nccl
process_count
int

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

Valeur par défaut: None
node_count
int

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

Valeur par défaut: 1