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	Optional[int] Número de tareas del servidor de parámetros. El valor predeterminado es 0.
worker_count	Optional[int] 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.