TensorFlowDistribution 클래스

TensorFlow 배포 구성.

상속

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

TensorFlowDistribution

생성자

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

키워드 전용 매개 변수

Name	Description
parameter_server_count	Optional[int] 매개 변수 서버 작업의 수입니다. 기본값은 0입니다.
worker_count	Optional[int] 작업자 수입니다. 기본값은 instance 개수입니다.

예제

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

변수

Name	Description
parameter_server_count	int 매개 변수 서버 작업 수입니다.
worker_count	int 작업자 수입니다. 지정하지 않으면 는 기본적으로 instance 개수로 설정됩니다.
type	str 배포 유형을 지정합니다. 이 클래스에 대해 자동으로 "tensorflow"로 설정합니다.