MCPcopy Create free account
hub / github.com/tensorflow/lingvo / GetExecutorParams

Method GetExecutorParams

lingvo/trainer.py:280–299  ·  view source on GitHub ↗

Get the params needed to instantiate the ExecutorTpu. Args: executor_type: either 'executor_tpu' or 'host_driven_executor_tpu'. Returns: Tuple (dict, params): - ps_params_dict: high_level task_name -> ProgramScheduleParams - train_cfg: Either a SingleTaskMod

(
      self, executor_type: Literal['executor_tpu', 'host_driven_executor_tpu']
  )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_CreateRunnerMethod · 0.95

Calls 2

CurrentMethod · 0.80

Tested by

no test coverage detected