Skip to content

Instantly share code, notes, and snippets.

@deepak-karkala

deepak-karkala/tf_model_saving.py

Created December 20, 2020 19:30
Show Gist options
  • Save deepak-karkala/4b88a079afdf636ff6358bd9f59d6383 to your computer and use it in GitHub Desktop.
Save deepak-karkala/4b88a079afdf636ff6358bd9f59d6383 to your computer and use it in GitHub Desktop.
Download ZIP
Saving Tensorflow models for serving
Raw
tf_model_saving.py
# Saving Tensorflow models for serving
# Option 1: SavedModel format
# Save model
model.save('saved_model/model')
# Load model
loaded_model = tf.keras.models.load_model('saved_model/model')
# Option 2: .h5 format
# Save model
save_model_path = "/tmp/model.h5"
model.save(save_model_path)
# Load model
loaded_model = keras.models.load_model(save_model_path)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment