add config.enable_memory_optim()

160bb06e · MissPenguin · GitHub · f051be86 · 160bb06e
Unverified Commit 160bb06e authored Jun 23, 2020 by MissPenguin Committed by GitHub Jun 23, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tools/infer/utility.py tools/infer/utility.py +2 -1

No files found.
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -88,6 +88,7 @@ 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()