View source on GitHub |
Gets train actions for TFM trainer.
tfm.core.actions.get_train_actions(
params: tfm.core.base_trainer.ExperimentConfig
,
trainer: tfm.core.base_trainer.Trainer
,
model_dir: str,
checkpoint_manager: tf.train.CheckpointManager
) -> List[orbit.Action]