Module: tfa.callbacks

View source on GitHub

Additional callbacks that conform to Keras API.

Modules

average_model_checkpoint module

time_stopping module: Callback that stops training when a specified amount of time has passed.

tqdm_progress_bar module: TQDM Progress Bar.

Classes

class AverageModelCheckpoint: The callback that should be used with optimizers that extend

class TQDMProgressBar: TQDM Progress Bar for Tensorflow Keras.

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