How to save checkpoint in between each epochs and retain them after and continue?

how to save checkpoint in Google colab while training for tflite , save checkpoint after each epochs and continue after…

Example
Google Colab