未验证 提交 e4600832 编写于 作者: D Double_V 提交者: GitHub

Update utility.py

上级 30d0cc1f
......@@ -316,6 +316,10 @@ def get_output_tensors(args, mode, predictor):
output_name = 'softmax_0.tmp_0'
if output_name in output_names:
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:
for output_name in output_names:
output_tensor = predictor.get_output_handle(output_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册