tf.compat.v1.saved_model.load
Загружает модель из SavedModel, как указано метками. (устарело)
tf.compat.v1.saved_model.load(
sess, tags, export_dir, import_scope=None, **saver_kwargs
)
Миграция на TF2
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"])
Описание
Используется в блокнотах
| Используется в руководстве |
|---|
| Аргументы | |
|---|---|
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