Merge pull request #5950 from LDOUBLEV/dygraph

fix rec output name error

Merge pull request #5950 from LDOUBLEV/dygraph
fix rec output name error
e4ab0ebe · andyjpaddle · GitHub · a8318353 · 243f89da · e4ab0ebe
显示空白变更内容
内联并排

Showing with 4 addition and 0 deletion

tools/infer/utility.py tools/infer/utility.py +4 -0

未找到文件。
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -327,6 +327,10 @@ def get_output_tensors(args, mode, predictor):
            for output_name in output_names:
                output_tensor = predictor.get_output_handle(output_name)
                output_tensors.append(output_tensor)
+    else:
+        for output_name in output_names:
+            output_tensor = predictor.get_output_handle(output_name)
+            output_tensors.append(output_tensor)
    return output_tensors