當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.compat.v1.saved_model.load用法及代碼示例


從標簽指定的 SavedModel 加載模型。 (已棄用)

用法

tf.compat.v1.saved_model.load(
    sess, tags, export_dir, import_scope=None, **saver_kwargs
)

參數

  • sess 用於恢複變量的 TensorFlow 會話。
  • tags 一組字符串標簽，用於標識所需的 MetaGraphDef。這些應與使用 SavedModel save() API 保存變量時使用的標簽相對應。
  • export_dir SavedModel 協議緩衝區和要加載的變量所在的目錄。
  • import_scope 可選的string-- 如果指定，在所有加載的張量名稱前麵加上 '/'。此範圍適用於加載到傳遞的會話中的張量實例，但它是不是寫入靜態MetaGraphDef返回的協議緩衝區。
  • **saver_kwargs 傳遞給 Saver 的可選關鍵字參數。

返回

  • 在提供的會話中加載的 MetaGraphDef 協議緩衝區。這可用於進一步提取signature-defs、collection-defs等。

拋出

  • RuntimeError 找不到與標簽關聯的 MetaGraphDef。

遷移到 TF2

警告：這個 API 是為 TensorFlow v1 設計的。繼續閱讀有關如何從該 API 遷移到本機 TensorFlow v2 等效項的詳細信息。見TensorFlow v1 到 TensorFlow v2 遷移指南有關如何遷移其餘代碼的說明。

tf.compat.v1.saved_model.load 或 tf.compat.v1.saved_model.loader.load 與即刻執行不兼容。請改用tf.saved_model.load 來加載您的模型。您可以參考 SavedModel 指南以獲取更多信息以及 tf.saved_model.load 文檔字符串中的“從 TensorFlow 1.x 導入 SavedModels”。

如何映射參數

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"])

警告：此函數已棄用。它將在未來的版本中刪除。更新說明：此函數隻能通過 v1 兼容庫作為 tf.compat.v1.saved_model.loader.load 或 tf.compat.v1.saved_model.load 使用。在 Tensorflow 2.0 中將有一個新的函數用於導入 SavedModels。

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.compat.v1.saved_model.load。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。