Skip to content

Instantly share code, notes, and snippets.

@Sirsirious

Sirsirious/save_model_keras.py

Created December 14, 2020 17:52
Show Gist options
  • Save Sirsirious/2f59f8227e83af42d2459fa88ed9b6de to your computer and use it in GitHub Desktop.
Save Sirsirious/2f59f8227e83af42d2459fa88ed9b6de to your computer and use it in GitHub Desktop.
Download ZIP
Raw
save_model_keras.py
model_file = os.path.join("/content/", "saved_model")
# This yields a .pb file in the defined path: /content/saved_model/saved_model.pb
keras_model.save(model_file)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment