Thanks! This is exactly what I’m looking for.
Is there a way to average over just the last N epochs instead of all of the epochs?
If I train the model normally and then call model.fit again with an AverageModelCheckpoint, then will it average over all epochs or just the ones from the second call to model.fit?