ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfa.callbacks

Additional callbacks that conform to Keras API.


class AverageModelCheckpoint: The callback that saves average model weights.

class TQDMProgressBar: TQDM Progress Bar for Tensorflow Keras.

class TimeStopping: Stop training when a specified amount of time has passed.