Compartir a través de


TensorFlowDistribution Clase

Configuración de distribución de TensorFlow.

Herencia
azure.ai.ml.entities._job.distribution.DistributionConfiguration
TensorFlowDistribution

Constructor

TensorFlowDistribution(*, parameter_server_count: int | None = 0, worker_count: int | None = None, **kwargs: Any)

Parámetros de palabra clave únicamente

Nombre Description
parameter_server_count

Número de tareas del servidor de parámetros. El valor predeterminado es 0.

worker_count

Número de trabajadores. El valor predeterminado es el recuento de instancias.

Ejemplos

Configuración de un commandComponent con 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

Nombre Description
parameter_server_count
int

Número de tareas del servidor de parámetros.

worker_count
int

Número de trabajadores. Si no se especifica, el valor predeterminado será el recuento de instancias.

type
str

Especifica el tipo de distribución. Establezca automáticamente en "tensorflow" para esta clase.