fix feed type in cpp_infer.py (#170)

256c2c9c · wangguanzhong · GitHub · 820a38b6 · 256c2c9c
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

tools/cpp_infer.py tools/cpp_infer.py +2 -2

未找到文件。
--- 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: