tfa.callbacks.AverageModelCheckpoint

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.

Methods

set_model

View source

set_params