fix new docker error

Change-Id: Ib47ef63da3ed99849d841d0df24c6c15952c672e
sophgo · Feb 15, 2023 · 57f4a87 · 57f4a87
1 parent 123549b
commit 57f4a87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tools/model_runner.py b/python/tools/model_runner.py
@@ -229,7 +229,7 @@ def generate_onnx_with_all(onnx_file: str):
         output_num = len(outs) - len(output_keys)
         outs = outs[output_num:]
         os.remove(onnx_file)
-        return dict(zip(output_keys, map(np.ndarray.flatten, outs)))
+        return dict(zip(output_keys, outs))
 
 
 def caffe_inference(inputs: dict, prototxt: str, caffemodel: str, dump_all: bool = True) -> dict: