Module: tfa.callbacks

Additional callbacks that conform to Keras API.

Classes

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.