Python tf.saved_model.Asset用法及代碼示例

表示要密封地包含在 SavedModel 中的文件資產。

用法

tf.saved_model.Asset(
    path
)

SavedModel 可以包含其使用所需的任意文件，稱為資產。例如，用於初始化查找表的詞匯文件。

當通過 tf.saved_model.save() 導出可跟蹤對象時，所有可從中訪問的 Asset 都將複製到 SavedModel 資產目錄中。加載時，資產和依賴於它們的序列化函數將引用 SavedModel 目錄中的正確文件路徑。

例子：

filename = tf.saved_model.Asset("file.txt")

@tf.function(input_signature=[])
def func():
  return tf.io.read_file(filename)

trackable_obj = tf.train.Checkpoint()
trackable_obj.func = func
trackable_obj.filename = filename
tf.saved_model.save(trackable_obj, "/tmp/saved_model")

# The created SavedModel is hermetic, it does not depend on
# the original file and can be moved to another path.
tf.io.gfile.remove("file.txt")
tf.io.gfile.rename("/tmp/saved_model", "/tmp/new_location")

reloaded_obj = tf.saved_model.load("/tmp/new_location")
print(reloaded_obj.func())