fix some args positon (#1228)

fix server and client var
This commit is contained in:
shaohua.zhang 2020-11-26 21:36:22 +08:00 committed by GitHub
parent ca5308c94e
commit e6f89bd85e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions

View File

@ -24,9 +24,9 @@ def parse_args():
args = parse_args()
inference_model_dir = args.model_dir
serving_client_dir = os.path.join(args.model_dir, args.server_dir)
serving_server_dir = os.path.join(args.model_dir, args.client_dir)
serving_server_dir = os.path.join(args.model_dir, args.server_dir)
serving_client_dir = os.path.join(args.model_dir, args.client_dir)
feed_var_names, fetch_var_names = inference_model_to_serving(
inference_model_dir, serving_client_dir, serving_server_dir, model_filename="model", params_filename="params")
inference_model_dir, serving_server_dir, serving_client_dir, model_filename="model", params_filename="params")
print("success!")