Skip to content

GitLab

Commit 17f2f812 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 336229644
parent dc2ae42c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/serving/export_saved_model.py
View file @ 17f2f812
...@@ -32,7 +32,7 @@ export_dir_path = XX ...@@ -32,7 +32,7 @@ export_dir_path = XX
input_type = XX input_type = XX
input_images = XX input_images = XX
imported = tf.saved_model.load(export_dir_path) imported = tf.saved_model.load(export_dir_path)
model_fn = .signatures['serving_default'] model_fn = imported.signatures['serving_default']
output = model_fn(input_images) output = model_fn(input_images)
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment