TensorFlowDistribution Classe
Configuration de distribution TensorFlow.
- Héritage
-
azure.ai.ml.entities._job.distribution.DistributionConfigurationTensorFlowDistribution
Constructeur
TensorFlowDistribution(*, parameter_server_count: int | None = 0, worker_count: int | None = None, **kwargs: Any)
Paramètres de mot clé uniquement
Nom | Description |
---|---|
parameter_server_count
|
Nombre de tâches du serveur de paramètres. La valeur par défaut est 0. |
worker_count
|
Nombre de travailleurs. Par défaut, le nombre de instance. |
Exemples
Configuration d’un CommandComponent avec un TensorFlowDistribution.
from azure.ai.ml import TensorFlowDistribution
from azure.ai.ml.entities import CommandComponent
component = CommandComponent(
name="microsoftsamples_tf",
description="This is the TF command component",
inputs={
"component_in_number": {"description": "A number", "type": "number", "default": 10.99},
"component_in_path": {"description": "A path", "type": "uri_folder"},
},
outputs={"component_out_path": {"type": "uri_folder"}},
command="echo Hello World & echo ${{inputs.component_in_number}} & echo ${{inputs.component_in_path}} "
"& echo ${{outputs.component_out_path}}",
environment="AzureML-sklearn-1.0-ubuntu20.04-py38-cpu:33",
distribution=TensorFlowDistribution(
parameter_server_count=1,
worker_count=2,
),
instance_count=2,
)
Variables
Nom | Description |
---|---|
parameter_server_count
|
Nombre de tâches de serveur de paramètres. |
worker_count
|
Nombre de workers. S’il n’est pas spécifié, est défini par défaut sur le nombre de instance. |
type
|
Spécifie le type de distribution. Définissez automatiquement sur « tensorflow » pour cette classe. |
Collaborer avec nous sur GitHub
La source de ce contenu se trouve sur GitHub, où vous pouvez également créer et examiner les problèmes et les demandes de tirage. Pour plus d’informations, consultez notre guide du contributeur.
Azure SDK for Python