Merge branch 'dygraph' into fix_attn_export

2021-03-12 11:06:04 +08:00 · 2021-03-12 11:06:04 +08:00 · c2449bded2
parent b0e04e06ce 5eafd4672c
commit c2449bded2
1 changed files with 2 additions and 1 deletions
--- 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")