flytekitplugins.kftensorflow.TfJob

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

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

Parameters:

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

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

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

  • evaluator (Evaluator) – Configuration for the evaluator replica group.

  • run_policy (RunPolicy | None) – Configuration for the run policy.

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

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

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

  • num_evaluator_replicas (int | None)

Methods

Attributes

num_chief_replicas: int | None = None
num_evaluator_replicas: int | None = None
num_ps_replicas: int | None = None
num_workers: int | None = None
chief: Chief
ps: PS
worker: Worker
evaluator: Evaluator
run_policy: RunPolicy | None