Модуль: tf.compat.v1.saved_model.loader

Функциональность загрузчика для SavedModel с герметичными, независимыми от языка экспортами.

Возможность загрузки и восстановления SavedModel, который может включать несколько мета-графических определений. Каждый SavedModel связан с одним контрольным пунктом. Каждое мета-графическое определение сохраняется с одним или несколькими тегами, которые используются для идентификации точного мета-графического определения для загрузки.

Операция load требует сессии, в которой должно быть восстановлено определение графика и переменные, теги для идентификации мета-графического определения для загрузки и местоположения SavedModel.

При загрузке подмножество переменных и активов, предоставленных в рамках конкретного мета-графического определения, будет восстановлено в предоставленную сессию. Однако значения переменных будут соответствовать сохранённым значениям из первого мета-графического определения, добавленного в SavedModel с использованием add_meta_graph_and_variables(...) в builder.py.

Пример использования:

...
builder = tf.compat.v1.saved_model.builder.SavedModelBuilder(export_dir)

with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  ...
  builder.add_meta_graph_and_variables(sess,
                                       ["foo-tag"],
                                       signature_def_map=foo_signatures,
                                       assets_collection=foo_assets)
...

with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  ...
  builder.add_meta_graph(["bar-tag", "baz-tag"],
                         assets_collection=bar_baz_assets)
...

builder.save()

...
with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  tf.compat.v1.saved_model.loader.load(sess, ["foo-tag"], export_dir)
  ...

Функции

load(...): Загружает модель из SavedModel, как указано тегами. (устарело)

maybe_saved_model_directory(...): Проверяет, может ли предоставленная директория экспорта содержать SavedModel.