TensorFlowDistribution Classe
Configuração de distribuição do TensorFlow.
- Herança
-
azure.ai.ml.entities._job.distribution.DistributionConfigurationTensorFlowDistribution
Construtor
TensorFlowDistribution(*, parameter_server_count: int | None = 0, worker_count: int | None = None, **kwargs: Any)
Parâmetros somente de palavra-chave
Nome | Description |
---|---|
parameter_server_count
|
O número de tarefas do servidor de parâmetros. Assume o padrão de 0. |
worker_count
|
O número de trabalhadores. O padrão é a contagem de instâncias. |
Exemplos
Configurando um CommandComponent com uma 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,
)
Variáveis
Nome | Description |
---|---|
parameter_server_count
|
Número de tarefas do servidor de parâmetros. |
worker_count
|
Número de funções de trabalho. Se não for especificado, o padrão será a contagem de instâncias. |
type
|
Especifica o tipo de distribuição. Defina automaticamente como "tensorflow" para essa classe. |
Colaborar conosco no GitHub
A fonte deste conteúdo pode ser encontrada no GitHub, onde você também pode criar e revisar problemas e solicitações de pull. Para obter mais informações, confira o nosso guia para colaboradores.
Azure SDK for Python