From 256c2c9c063d5ae1370ab8aad8f643c0d862962b Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Thu, 9 Jan 2020 14:38:50 +0800 Subject: [PATCH] fix feed type in cpp_infer.py (#170) --- tools/cpp_infer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/cpp_infer.py b/tools/cpp_infer.py index 40980f6ad..41b1e8a00 100644 --- a/tools/cpp_infer.py +++ b/tools/cpp_infer.py @@ -220,7 +220,7 @@ def infer(): for i in range(10): if conf['use_python_inference']: outs = exe.run(infer_prog, - feed=[data_dict], + feed=data_dict, fetch_list=fetch_targets, return_numpy=False) else: @@ -232,7 +232,7 @@ def infer(): for i in range(cnt): if conf['use_python_inference']: outs = exe.run(infer_prog, - feed=[data_dict], + feed=data_dict, fetch_list=fetch_targets, return_numpy=False) else: -- GitLab