未验证 提交 b27acf6a 编写于 作者: T Tingquan Gao 提交者: GitHub

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.
上级 7abf42b7
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册