Модуль: 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.