Spec-Zone.ru › TensorFlow

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 в эквивалент 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"])

Описание

Используется в блокнотах

Используется в руководстве
  • Мигрируйте рабочий процесс SavedModel
Устарело: ЭТА ФУНКЦИЯ УСТАРЕЛА. Она будет удалена в будущей версии. Инструкции по обновлению: Используйте tf.saved_model.load вместо этого.
Аргументы
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/api_docs/python/tf/compat/v1/saved_model/load

Spec-Zone.ru

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