tf.saved_model.loader.load

tf.saved_model.loader.load(
    sess,
    tags,
    export_dir,
    **saver_kwargs
)

Defined in tensorflow/python/saved_model/loader_impl.py.

Loads the model from a SavedModel as specified by tags.

Args:

  • sess: The TensorFlow session to restore the variables.
  • tags: Set of string tags to identify the required MetaGraphDef. These should correspond to the tags used when saving the variables using the SavedModel save() API.
  • export_dir: Directory in which the SavedModel protocol buffer and variables to be loaded are located.
  • **saver_kwargs: Optional keyword arguments passed through to Saver.

Returns:

The MetaGraphDef protocol buffer loaded in the provided session. This can be used to further extract signature-defs, collection-defs, etc.

Raises:

  • RuntimeError: MetaGraphDef associated with the tags cannot be found.