你当前正在访问 Microsoft Azure Global Edition 技术文档网站。 如果需要访问由世纪互联运营的 Microsoft Azure 中国技术文档网站，请访问 https://docs.azure.cn。

TensorFlowDistribution 类

  • 参考

TensorFlow 分发配置。

继承
azure.ai.ml.entities._job.distribution.DistributionConfiguration
TensorFlowDistribution

构造函数

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

仅限关键字的参数

名称 说明
parameter_server_count
Optional[int]

参数服务器任务数。 默认值为 0。
worker_count
Optional[int]

辅助角色数。 默认为实例计数。

示例

使用 TensorFlowDistribution 配置 CommandComponent。


   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,
   )

变量

名称 说明
parameter_server_count
int

参数服务器任务数。
worker_count
int

工作线程数。 如果未指定，则默认为实例计数。
type
str

指定分布的类型。 此类自动设置为“tensorflow”。