flytekitplugins.kftensorflow.TfJob#

class flytekitplugins.kftensorflow.TfJob(chief=<factory>, ps=<factory>, worker=<factory>, run_policy=<factory>, num_workers=None, num_ps_replicas=None, num_chief_replicas=None)[source]#

Configuration for an executable TensorFlow Job. Use this to run distributed TensorFlow training on Kubernetes.

Parameters

  • chief (flytekitplugins.kftensorflow.task.Chief) – Configuration for the chief replica group.

  • ps (flytekitplugins.kftensorflow.task.PS) – Configuration for the parameter server (PS) replica group.

  • worker (flytekitplugins.kftensorflow.task.Worker) – Configuration for the worker replica group.

  • run_policy (Optional[flytekitplugins.kftensorflow.task.RunPolicy]) – Configuration for the run policy.

  • num_workers (Optional[int]) – [DEPRECATED] This argument is deprecated. Use worker.replicas instead.

  • num_ps_replicas (Optional[int]) – [DEPRECATED] This argument is deprecated. Use ps.replicas instead.

  • num_chief_replicas (Optional[int]) – [DEPRECATED] This argument is deprecated. Use chief.replicas instead.

Return type

None

Methods

Attributes

num_chief_replicas: Optional[int] = None
num_ps_replicas: Optional[int] = None
num_workers: Optional[int] = None
chief: flytekitplugins.kftensorflow.task.Chief
ps: flytekitplugins.kftensorflow.task.PS
worker: flytekitplugins.kftensorflow.task.Worker
run_policy: Optional[flytekitplugins.kftensorflow.task.RunPolicy]