|View source on GitHub|
Keeps track of the best result, and saves its checkpoint.
tfm.core.train_utils.BestCheckpointExporter( export_dir: str, metric_name: str, metric_comp: str )
Orbit will support an API for checkpoint exporter. This class will be used together with orbit once this functionality is ready.
||Returns the best ckpt path or None if there is no ckpt yet.|
export_best_eval_metric( eval_logs, global_step )
Export evaluation results of the best checkpoint into a json file.
maybe_export_checkpoint( checkpoint, eval_logs, global_step, write_logs=True ) -> bool
Compare eval_logs with past eval_logs and export checkpoint if better.