Skip to content

Commit

Permalink
Include link to SavedModel CLI in the Py loader when tag-set is not f…
Browse files Browse the repository at this point in the history
…ound.

PiperOrigin-RevId: 159586182
  • Loading branch information
sukritiramesh authored and tensorflower-gardener committed Jun 20, 2017
1 parent d4fd969 commit fa7f73d
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tensorflow/python/saved_model/loader_impl.py
Expand Up @@ -206,8 +206,11 @@ def load(sess, tags, export_dir, **saver_kwargs):
break

if not found_match:
raise RuntimeError("MetaGraphDef associated with tags " + str(tags).strip(
"[]") + " could not be found in SavedModel")
raise RuntimeError(
"MetaGraphDef associated with tags " + str(tags).strip("[]") +
" could not be found in SavedModel. To inspect available tag-sets in"
" the SavedModel, please use the SavedModel CLI: `saved_model_cli`"
)

# Build a saver by importing the meta graph def to load.
saver = tf_saver.import_meta_graph(meta_graph_def_to_load, **saver_kwargs)
Expand Down

0 comments on commit fa7f73d

Please sign in to comment.