Module: tfm.core.base_trainer

Stay organized with collections Save and categorize content based on your preferences.

Standard Trainer implementation.

The base trainer implements the Orbit StandardTrainable and StandardEvaluable interfaces. Trainers inside this project should be interchangable and independent on model architectures and tasks.


class ExperimentConfig: Top-level configuration.

class Trainer: Implements the common trainer shared for TensorFlow models.

class TrainerConfig: Configuration for trainer.


get_runtime_options(...): Get tf.distribute.RunOptions from config.