Warning: This project is deprecated. TensorFlow Addons has stopped development, The project will only be providing minimal maintenance releases until May 2024. See the full announcement here or on github.


The callback that saves average model weights.

Used in the notebooks

Used in the tutorials

The callback that should be used with optimizers that extend tfa.optimizers.AveragedOptimizerWrapper, i.e., tfa.optimizers.MovingAverage and tfa.optimizers.StochasticAverage optimizers. It saves and, optionally, assigns the averaged weights.

update_weights If True, assign the moving average weights to the model, and save them. If False, keep the old non-averaged weights, but the saved model uses the average weights.

See tf.keras.callbacks.ModelCheckpoint for the other args.



View source