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/versions/r2.9/api_docs/python/tf/compat/v1/saved_model/load