From f78678dd66a798d78a8c9cb07af3c58dee7609b0 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Tue, 15 Sep 2020 14:23:41 +0000
Subject: [PATCH] fix py infer

---
 tools/infer/utility.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index 8d66f867..1d07f62f 100755
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -101,6 +101,7 @@ def create_predictor(args, mode):
         config.disable_gpu()
         config.set_cpu_math_library_num_threads(6)
         if args.enable_mkldnn:
+            config.set_mkldnn_cache_capacity(10)
             config.enable_mkldnn()
 
     #config.enable_memory_optim()
-- 
GitLab