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

構建 SavedModel 協議緩衝區並保存變量和資產。

用法

tf.compat.v1.saved_model.Builder(
    export_dir
)

SavedModelBuilder 類提供了構建SavedModel 協議緩衝區的函數。具體來說，這允許將多個元圖保存為單個 language-neutral SavedModel 的一部分，同時共享變量和資產。

要構建 SavedModel，第一個元圖必須與變量一起保存。隨後的元圖將與其圖定義一起保存。如果資產需要保存並寫入或複製到磁盤，可以在添加元圖def時提供。如果多個元圖定義與同名資產相關聯，則僅保留第一個版本。

添加到 SavedModel 的每個元圖都必須使用標簽進行注釋。標簽提供了一種方法來識別要加載和恢複的特定元圖，以及共享的變量和資產集。

SavedModelBuilder 的典型用法：

...
builder = tf.compat.v1.saved_model.Builder(export_dir)

with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  ...
  builder.add_meta_graph_and_variables(sess,
                                  ["foo-tag"],
                                  signature_def_map=foo_signatures,
                                  assets_collection=foo_assets)
...

with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  ...
  builder.add_meta_graph(["bar-tag", "baz-tag"])
...

builder.save()

注意：此函數隻能通過 v1 兼容性庫作為 tf.compat.v1.saved_model.builder.SavedModelBuilder 或 tf.compat.v1.saved_model.Builder 使用。 Tensorflow 2.0 將引入一種基於對象的新方法來創建 SavedModel。