Tensorflow Create .ckpt Files Instead of Folders

Photo by James Harrison on Unsplash

Tensorflow An end-to-end open source machine learning platform.

Some Basics about saving and loading the model:

Saving a Keras model:

model = … # Get model (Sequential, Functional Model, or Model subclass)
model.save('path/to/location')

Loading the model back:

from tensorflow import keras
model = keras.models.load_model('path/to/location')

While working on a tensorflow project, I was getting folders and not .cpkt files. So i added save_weights_only=True,

checkpoint_callback = tf.keras.callbacks.ModelCheckpoint(
    filepath=checkpoint_path,
    verbose=1,
    save_weights_only=True,
    period=2
)

as a paramter while saving the model and it started making the .cpkt files.


Author Images
Syed
From Islamabad

A software developer with a keen interest in writing about technology, finance, and entrepreneurship. I've written for businesses in a variety of fields, including new technology, healthcare, programming, consumer applications, corporate computing, UI/UX, outsourcing, and education.

Post a comment

Your email address will not be published. Required fields are marked *

    Press x to close