flytekitplugins.kftensorflow.TfJob

class flytekitplugins.kftensorflow.TfJob(num_workers, num_ps_replicas, num_chief_replicas)

Configuration for an executable TF Job. Use this to run distributed tensorflow training on k8s (with parameter server)

Parameters

• num_workers (int) – integer determining the number of worker replicas spawned in the cluster for this job

• master). ((in addition to 1) –

• num_ps_replicas (int) – Number of Parameter server replicas to use

• num_chief_replicas (int) – Number of chief replicas to use

Return type

None

Methods

Attributes

num_workers: int

num_ps_replicas: int

num_chief_replicas: int