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 execution. Для загрузки модели используйте 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, загруженный в предоставленный сеанс. Его можно использовать для дополнительного извлечения описаний сигнатур, описаний коллекций и т. д.
Вызывает исключение
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