How can I save the best model with a minimun epoch?

[xianyuanliu]

Hi all,

In my case, I hope model training for 10 epochs before adding an extra loss, and I need to get the best model for the rest epochs. Can I use ModelCheckpoint to do it? Many thanks!