From 88b0a3796f8f77b3d21e70a700dec1e6894c0fe1 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Fri, 3 Apr 2020 10:29:49 +0800 Subject: [PATCH] add memory optimize for cpu server --- python/paddle_serving_server/serve.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/paddle_serving_server/serve.py b/python/paddle_serving_server/serve.py index 279e3a89..088e3928 100644 --- a/python/paddle_serving_server/serve.py +++ b/python/paddle_serving_server/serve.py @@ -38,6 +38,8 @@ def parse_args(): # pylint: disable=doc-string-missing help="Working dir of current service") parser.add_argument( "--device", type=str, default="cpu", help="Type of device") + parser.add_argument( + "--mem_optim", type=bool, default=False, help="Memory optimize") return parser.parse_args() @@ -48,6 +50,7 @@ def start_standard_model(): # pylint: disable=doc-string-missing port = args.port workdir = args.workdir device = args.device + mem_optim = args.mem_optim if model == "": print("You must specify your serving model") @@ -67,6 +70,7 @@ def start_standard_model(): # pylint: disable=doc-string-missing server = serving.Server() server.set_op_sequence(op_seq_maker.get_op_sequence()) server.set_num_threads(thread_num) + server.set_memory_optimize(mem_optim) server.load_model_config(model) server.prepare_server(workdir=workdir, port=port, device=device) -- GitLab