Spec-Zone.ru › TensorFlow 2.9

tf.compat.v1.saved_model.load

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

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.saved_model.loader.load

tf.compat.v1.saved_model.load(
    sess, tags, export_dir, import_scope=None, **saver_kwargs
)

Миграция на TF2

Внимание: Этот API был разработан для TensorFlow v1. Продолжайте чтение для получения подробной информации о миграции с этого API на эквивалентный API TensorFlow v2. См. Руководство по миграции TensorFlow v1 в TensorFlow v2 для получения инструкций по миграции остальной части кода.

tf.compat.v1.saved_model.load или tf.compat.v1.saved_model.loader.load несовместим с выполнением eager. Используйте tf.saved_model.load для загрузки модели вместо этого. Вы можете обратиться к Руководству SavedModel для получения дополнительной информации, а также к разделу «Импорт SavedModels из TensorFlow 1.x» в tf.saved_model.load документации.

Как сопоставить аргументы

Имя аргумента TF1 Имя аргумента TF2 Примечание
sess Не поддерживается -
tags tags -
export_dir export_dir -
import_scope Не поддерживается Имена областей не требуются. По умолчанию переменные ассоциируются с загруженным объектом, а имена функций дублируются.
saver_kwargs Не поддерживается -

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

До:

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

После:

model = tf.saved_model.load(export_dir, tags=["foo-tag"])

Описание

Устарело: ЭТА ФУНКЦИЯ УСТАРЕЛА. Она будет удалена в будущей версии. Инструкции по обновлению: Эта функция будет доступна только через библиотеку совместимости v1 как tf.compat.v1.saved_model.loader.load или tf.compat.v1.saved_model.load. В Tensorflow 2.0 будет новая функция для импорта SavedModels.
Аргументы
sess Сессия TensorFlow для восстановления переменных.
tags Набор строковых тегов для идентификации требуемого MetaGraphDef. Они должны соответствовать тегам, используемым при сохранении переменных с помощью API SavedModel save()
export_dir Директория, в которой расположены SavedModel протокол буфер и переменные, подлежащие загрузке.
import_scope Необязательный string — если указано, префикс этой строки, за которым следует '/', к всем загруженным именам тензоров. Эта область применяется к экземплярам тензоров, загруженным в переданную сессию, но не записывается в статический MetaGraphDef протокол буфер, который возвращается.
**saver_kwargs Необязательные ключевые аргументы, передаваемые через Saver.
Возвращает
Загруженный в предоставленной сессии MetaGraphDef протокол буфер. Это можно использовать для дальнейшего извлечения signature-defs, collection-defs и т. д.
Возбуждает
RuntimeError MetaGraphDef, связанный с тегами, не найден.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/saved_model/load

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API