Module: tfa.callbacks

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

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.