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.pull/1683/head
parent
7abf42b7a1
commit
b27acf6ace
|
@ -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)
|
||||
|
||||
|
|
Loading…
Reference in New Issue