From ff253cae369b91573fa19fcd30320523313b9fa5 Mon Sep 17 00:00:00 2001 From: Tingquan Gao Date: Mon, 28 Feb 2022 09:53:49 +0800 Subject: [PATCH] fix: remove set_mkldnn_cache_capacity (#1695) use set_mkldnn_cache_capacity() to avoid memory leak and the bug is fixed in Paddle so that it is not needed in cv now. --- deploy/utils/predictor.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/deploy/utils/predictor.py b/deploy/utils/predictor.py index d44ae03d..b3bce5d8 100644 --- a/deploy/utils/predictor.py +++ b/deploy/utils/predictor.py @@ -50,8 +50,6 @@ class Predictor(object): else: config.disable_gpu() if args.enable_mkldnn: - # cache 10 different shapes for mkldnn to avoid memory leak - config.set_mkldnn_cache_capacity(10) config.enable_mkldnn() config.set_cpu_math_library_num_threads(args.cpu_num_threads) -- GitLab