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())