Skip to content

Instantly share code, notes, and snippets.

@ywolff

ywolff/export_model_for_tf_serving.py

Created January 6, 2020 11:34
Show Gist options
  • Save ywolff/6cb903a567651874e2709a3f5e31aad8 to your computer and use it in GitHub Desktop.
Save ywolff/6cb903a567651874e2709a3f5e31aad8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
export_model_for_tf_serving.py
from pathlib import Path
import tensorflow as tf
from build_big_model import build_big_model
MODEL_NAME = 'big_model'
MODEL_VERSION = '1'
model = build_big_model()
tf.saved_model.save(model, str(Path('models') / MODEL_NAME / MODEL_VERSION))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment