Construct a HostDrivenExecutor BaseRunner. Args: train_cfg: SingleTaskModelParams or MultiTaskModelParams ps_params_dict: A dict of top-level task name -> ProgramSchedule params, if train_cfg is a SingleTaskModelParams, we expect only one entry. *args: List args to pas
(
self,
train_cfg: py_utils.InstantiableParams[
lingvo_program.HostDrivenTrainProgram],
ps_params_dict: Dict[str, lingvo_program.ProgramScheduleParamsT],
*args,
**kwargs,
)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected