flytekitplugins.kftensorflow.TfJob#

class flytekitplugins.kftensorflow.TfJob(num_workers, num_ps_replicas, num_chief_replicas)[source]#

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#