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

更新存儲在給定路徑中的 SavedModel 的方法名稱。

用法

tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater(
    export_dir
)

MethodNameUpdater 類提供了更新給定 SavedModel 的 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 使用。