experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add docstring to export_model

Browse Source
This commit is contained in:
Kamil Sindi 2018-01-04 15:58:34 -05:00
parent 81ec5d2093
commit ad8d437b32
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/examples/image_retraining/retrain.py
View File

@ -1009,6 +1009,13 @@ def add_jpeg_decoding(input_width, input_height, input_depth, input_mean,
def export_model(sess, architecture, saved_model_dir): def export_model(sess, architecture, saved_model_dir):
"""Exports model for serving.
Args:
sess: Current active TensorFlow Session.
architecture: Model architecture.
saved_model_dir: Directory in which to save exported model and variables.
"""
if architecture == 'inception_v3': if architecture == 'inception_v3':
input_tensor = 'DecodeJpeg/contents:0' input_tensor = 'DecodeJpeg/contents:0'
elif architecture.startswith('mobilenet_'): elif architecture.startswith('mobilenet_'):