TensorFlowDistribution Class

TensorFlow distribution configuration.

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

Constructor

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

Variables

parameter_server_count
int

Number of parameter server tasks.

worker_count
int

Number of workers. If not specified, will default to the instance count.

type
str

Specifies the type of distribution. Set automatically to "tensorflow" for this class.