TensorFlowDistribution Класс
Конфигурация распределения TensorFlow.
- Наследование
-
azure.ai.ml.entities._job.distribution.DistributionConfigurationTensorFlowDistribution
Конструктор
TensorFlowDistribution(*, parameter_server_count: int | None = 0, worker_count: int | None = None, **kwargs: Any)
Параметры Keyword-Only
Имя | Описание |
---|---|
parameter_server_count
|
Количество задач сервера параметров. Значение по умолчанию — 0. |
worker_count
|
Количество рабочих ролей. По умолчанию используется число экземпляров. |
Примеры
Настройка CommandComponent с 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,
)
Переменные
Имя | Описание |
---|---|
parameter_server_count
|
Число задач сервера параметров. |
worker_count
|
Количество рабочих ролей. Если значение не указано, по умолчанию будет использоваться число экземпляров. |
type
|
Указывает тип распределения. Автоматически задайте для этого класса значение tensorflow. |
Совместная работа с нами на GitHub
Источник этого содержимого можно найти на GitHub, где также можно создавать и просматривать проблемы и запросы на вытягивание. Дополнительные сведения см. в нашем руководстве для участников.
Azure SDK for Python
Обратная связь
https://aka.ms/ContentUserFeedback.
Ожидается в ближайшее время: в течение 2024 года мы постепенно откажемся от GitHub Issues как механизма обратной связи для контента и заменим его новой системой обратной связи. Дополнительные сведения см. в разделеОтправить и просмотреть отзыв по