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 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"])
Описание
Используется в блокнотах
| Используется в руководстве |
|---|
| Аргументы | |
|---|---|
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