tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater
Обновляет имя(а) метода сохранённой модели, расположенной по заданному пути.
tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater(
export_dir
)
Класс MethodNameUpdater предоставляет функциональность для обновления поля имени метода в signature_defs заданной сохранённой модели. Например, он может использоваться для замены predict method_name на regress.
Типичные примеры использования MethodNameUpdater
...
updater = tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater(
export_dir)
# Update all signature_defs with key "foo" in all meta graph defs.
updater.replace_method_name(signature_key="foo", method_name="regress")
# Update a single signature_def with key "bar" in the meta graph def with
# tags ["serve"]
updater.replace_method_name(signature_key="bar", method_name="classify",
tags="serve")
updater.save(new_export_dir)
Примечание: Эта функция будет доступна только через библиотеку совместимости v1 как tf.compat.v1.saved_model.builder.MethodNameUpdater.
| Аргументы | |
|---|---|
export_dir | Каталог, содержащий файлы SavedModel. |
| Исключения | |
|---|---|
IOError | Если файл сохранённой модели не существует или не может быть успешно проанализирован. |
Методы
replace_method_name
replace_method_name(
signature_key, method_name, tags=None
)
Заменяет метод_имя в указанной signature_def.
Это будет соответствовать и заменять несколько sig def, если tags равно None (т. е., когда несколько MetaGraph имеют signature_def с тем же ключом). Если tags не равно None, это заменит только одну signature_def в MetaGraph с совпадающими метками.
| Аргументы | |
|---|---|
signature_key | Ключ signature_def для обновления. |
method_name | Новое метод_имя для замены существующего. |
tags | Метка или последовательность меток, идентифицирующих MetaGraph для обновления. Если None, все метаграфы будут обновлены. |
| Исключения | |
|---|---|
ValueError | если ключ_подписи или метод_имя не определены или если не найдены метаграфы с соответствующими метками или если ни один метаграф не имеет signature_def, соответствующей ключу_подписи. |
save
save(
new_export_dir=None
)
Сохраняет обновлённую SavedModel.
| Аргументы | |
|---|---|
new_export_dir | Путь, где будет сохранён обновлённый SavedModel. Если None, входной SavedModel будет перезаписан с обновлениями. |
| Исключения | |
|---|---|
errors.OpError | Если при операции сохранения файла возникли ошибки. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/saved_model/signature_def_utils/MethodNameUpdater