From 6557ffe677fa5bc17c339b625884e913ebc656ed Mon Sep 17 00:00:00 2001 From: Jiawei Wang Date: Tue, 9 Jun 2020 17:00:27 +0800 Subject: [PATCH] fix api bug (#909) --- tools/export_serving_model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tools/export_serving_model.py b/tools/export_serving_model.py index 112acd552..07c9e7d77 100644 --- a/tools/export_serving_model.py +++ b/tools/export_serving_model.py @@ -50,8 +50,12 @@ def save_serving_model(FLAGS, exe, feed_vars, test_fetches, infer_prog): import paddle_serving_client.io as serving_io serving_client = os.path.join(save_dir, 'serving_client') serving_server = os.path.join(save_dir, 'serving_server') - serving_io.save_model(serving_client, serving_server, feed_dict, fetch_dict, - infer_prog) + serving_io.save_model( + client_config_folder=serving_client, + server_model_folder=serving_server, + feed_dict, + fetch_dict, + infer_prog) def main(): -- GitLab