From 47c90a1541a46236c9f2b7330b51e3507e0379fe Mon Sep 17 00:00:00 2001 From: MRXLT Date: Sun, 17 May 2020 17:13:46 +0800 Subject: [PATCH] refine inference model to serving --- doc/SAVE.md | 10 +++++----- doc/SAVE_CN.md | 10 +++++----- python/paddle_serving_client/io/__init__.py | 6 +++--- 3 files changed, 13 insertions(+), 13 deletions(-) diff --git a/doc/SAVE.md b/doc/SAVE.md index a1ff13f1..4fcdfa43 100644 --- a/doc/SAVE.md +++ b/doc/SAVE.md @@ -36,14 +36,14 @@ for line in sys.stdin: If you have saved model files using Paddle's `save_inference_model` API, you can use Paddle Serving's` inference_model_to_serving` API to convert it into a model file that can be used for Paddle Serving. ``` import paddle_serving_client.io as serving_io -serving_io.inference_model_to_serving(dirname, model_filename=None, params_filename=None, serving_server="serving_server", serving_client="serving_client") +serving_io.inference_model_to_serving(dirname, serving_server="serving_server", serving_client="serving_client", model_filename=None, params_filename=None ) ``` dirname (str) - Path of saved model files. Program file and parameter files are saved in this directory. -model_filename (str, optional) - The name of file to load the inference program. If it is None, the default filename `__model__` will be used. Default: None. - -paras_filename (str, optional) - The name of file to load all parameters. It is only used for the case that all parameters were saved in a single binary file. If parameters were saved in separate files, set it as None. Default: None. - serving_server (str, optional) - The path of model files and configuration files for server. Default: "serving_server". serving_client (str, optional) - The path of configuration files for client. Default: "serving_client". + +model_filename (str, optional) - The name of file to load the inference program. If it is None, the default filename `__model__` will be used. Default: None. + +paras_filename (str, optional) - The name of file to load all parameters. It is only used for the case that all parameters were saved in a single binary file. If parameters were saved in separate files, set it as None. Default: None. diff --git a/doc/SAVE_CN.md b/doc/SAVE_CN.md index de519da8..3ca715c0 100644 --- a/doc/SAVE_CN.md +++ b/doc/SAVE_CN.md @@ -37,14 +37,14 @@ for line in sys.stdin: 如果已使用Paddle 的`save_inference_model`接口保存出预测要使用的模型，则可以通过Paddle Serving的`inference_model_to_serving`接口转换成可用于Paddle Serving的模型文件。 ``` import paddle_serving_client.io as serving_io -serving_io.inference_model_to_serving(dirname, model_filename=None, params_filename=None, serving_server="serving_server", serving_client="serving_client") +serving_io.inference_model_to_serving(dirname, serving_server="serving_server", serving_client="serving_client", model_filename=None, params_filename=None) ``` dirname (str) – 需要转换的模型文件存储路径，Program结构文件和参数文件均保存在此目录。 -model_filename (str，可选) – 存储需要转换的模型Inference Program结构的文件名称。如果设置为None，则使用 `__model__` 作为默认的文件名。默认值为None。 - -params_filename (str，可选) – 存储需要转换的模型所有参数的文件名称。当且仅当所有模型参数被保存在一个单独的二进制文件中，它才需要被指定。如果模型参数是存储在各自分离的文件中，设置它的值为None。默认值为None。 - serving_server (str, 可选) - 转换后的模型文件和配置文件的存储路径。默认值为serving_server。 serving_client (str, 可选) - 转换后的客户端配置文件存储路径。默认值为serving_client。 + +model_filename (str，可选) – 存储需要转换的模型Inference Program结构的文件名称。如果设置为None，则使用 `__model__` 作为默认的文件名。默认值为None。 + +params_filename (str，可选) – 存储需要转换的模型所有参数的文件名称。当且仅当所有模型参数被保存在一个单独的二进制文件中，它才需要被指定。如果模型参数是存储在各自分离的文件中，设置它的值为None。默认值为None。 diff --git a/python/paddle_serving_client/io/__init__.py b/python/paddle_serving_client/io/__init__.py index 4f174866..93ae3705 100644 --- a/python/paddle_serving_client/io/__init__.py +++ b/python/paddle_serving_client/io/__init__.py @@ -104,10 +104,10 @@ def save_model(server_model_folder, def inference_model_to_serving(dirname, - model_filename=None, - params_filename=None, serving_server="serving_server", - serving_client="serving_client"): + serving_client="serving_client", + model_filename=None, + params_filename=None): place = fluid.CPUPlace() exe = fluid.Executor(place) inference_program, feed_target_names, fetch_targets = \ -- GitLab