How to save checkpoint while training tflite

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

Example

Hi @Sanju_k,

Incorporate the below code to save the model checkpoints.

import tensorflow as tf
from tensorflow.keras.callbacks import ModelCheckpoint
checkpoint_filepath = '/content/drive/MyDrive/checkpoints/model-{epoch}.h5'  # Replace with your desired path
checkpoint_callback = ModelCheckpoint(filepath=checkpoint_filepath, monitor='val_loss',  
    save_best_only=True,  verbose=1)

Thank You