Saving model checkpoint during the epoch

I would like to save a checkpoint every time a validation loop ends. I set up the val_check_interval to be 0.2 so I have 5 validation loops during each epoch but the checkpoint callback saves the model only at the end of the epoch. I couldn’t find an easy (or hard) way to save the model after each validation loop. It seems a bit strange cause, I think, the main purpose of the validation loop is to save a checkpoint.

What is the simplest way of doing that?

Thank you,

Gilad

It should checkpoint according to val_check_interval. Mind reproduce it with https://colab.research.google.com/drive/1HvWVVTK8j2Nj52qU4Q4YCyzOm0_aLQF3?usp=sharing??