use enable_memory_optim for cpu and gpu, add paddlehelp QRcode

2026-09-24 23:33:08 +08:00 · 2020-06-23 15:02:13 +08:00
parent 2401626ab5
commit b115cb4417
2 changed files with 1 additions and 1 deletions
@@ -88,10 +88,10 @@ def create_predictor(args, mode):

    if args.use_gpu:
        config.enable_use_gpu(args.gpu_mem, 0)
-        config.enable_memory_optim()
    else:
        config.disable_gpu()

+    config.enable_memory_optim()
    config.disable_glog_info()

    # use zero copy