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