Skip to content

GitLab

Unverified Commit 5eafd467 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #2230 from LDOUBLEV/fix_dyg 

enable memory optim
parents 0e332fb5 4c9d141c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tools/infer/utility.py
View file @ 5eafd467
...@@ -128,7 +128,8 @@ def create_predictor(args, mode, logger): ...@@ -128,7 +128,8 @@ def create_predictor(args, mode, logger):
#config.set_mkldnn_op({'conv2d', 'depthwise_conv2d', 'pool2d', 'batch_norm'}) #config.set_mkldnn_op({'conv2d', 'depthwise_conv2d', 'pool2d', 'batch_norm'})
args.rec_batch_num = 1 args.rec_batch_num = 1
# config.enable_memory_optim() # enable memory optim
config.enable_memory_optim()
config.disable_glog_info() config.disable_glog_info()
config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass") config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment