Module: tfm.core.train_lib

TFM common training driver library.

Classes

class OrbitExperimentRunner: Runs experiment with Orbit training loop.

Functions

maybe_create_best_ckpt_exporter(...): Maybe create a BestCheckpointExporter object, according to the config.

run_experiment(...): Runs train/eval configured by the experiment params.