Hi Marcus,
I guess for your use case (have the checkpoint and keep training later) I’d go straight to the regular Keras API. Create a simple model and do the transfer learning.
for this use case, there’s this tutorial here: Retraining an Image Classifier  |  TensorFlow Hub 3
It even shows how to convert the model to TFLite later.
hope it helps