Skip to content

Instantly share code, notes, and snippets.

@lakshmanok
Created August 28, 2018 16:12
Show Gist options
  • Save lakshmanok/107a18c1e83b4bfe9bed7893017d114c to your computer and use it in GitHub Desktop.
Save lakshmanok/107a18c1e83b4bfe9bed7893017d114c to your computer and use it in GitHub Desktop.
# export similar to Cloud ML Engine / TF Serving convention
tf.logging.info('Starting to export model.')
estimator.export_savedmodel(
export_dir_base=os.path.join(output_dir, 'export/exporter'),
serving_input_receiver_fn=serving_input_fn)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment