There was a feature - tensorflow format to save a checkpoint. I have checked and it is everywhere in the official docs and samples: https://www.tensorflow.org/tutorials/keras/save_and_load#savedmodel_format However in the TF I am running (2.18) if I write model.save('something') I get this: Invalid filepath extension for saving. Please add either a `.keras` extension for the native Keras format (recommended) or a `.h5` extension. How do I save in "tensorflow" format in the latest versions? Continue reading...