From 4c9d141c6f09913faa9f8ab54927b701b6f5bfef Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 11 Mar 2021 16:44:24 +0800 Subject: [PATCH] enable memory optim --- tools/infer/utility.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index a4a91efd..911ca7fc 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -128,7 +128,8 @@ def create_predictor(args, mode, logger): #config.set_mkldnn_op({'conv2d', 'depthwise_conv2d', 'pool2d', 'batch_norm'}) args.rec_batch_num = 1 - # config.enable_memory_optim() + # enable memory optim + config.enable_memory_optim() config.disable_glog_info() config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass") -- GitLab