Skip to content

GitLab

Commit b62e6954 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix rec output name error

parent 0b220341
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
tools/infer/utility.py
View file @ b62e6954
...@@ -323,6 +323,10 @@ def get_output_tensors(args, mode, predictor): ...@@ -323,6 +323,10 @@ def get_output_tensors(args, mode, predictor):
output_name = 'softmax_0.tmp_0' output_name = 'softmax_0.tmp_0'
if output_name in output_names: if output_name in output_names:
return [predictor.get_output_handle(output_name)] return [predictor.get_output_handle(output_name)]
else:
for output_name in output_names:
output_tensor = predictor.get_output_handle(output_name)
output_tensors.append(output_tensor)
else: else:
for output_name in output_names: for output_name in output_names:
output_tensor = predictor.get_output_handle(output_name) output_tensor = predictor.get_output_handle(output_name)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment