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.
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.
Post a comment