提交 30c54f99 编写于 作者: W wangjiawei04

fix imangnet pipeline err

上级 dd27f273
......@@ -52,10 +52,11 @@ class ImagenetOp(Op):
for score in score_list:
score = score.tolist()
max_score = max(score)
#result["label"].append(self.label_dict[score.index(max_score)]
#.strip().replace(",", ""))
#result["prob"].append(max_score)
#print(result)
result["label"].append(self.label_dict[score.index(max_score)]
.strip().replace(",", ""))
result["prob"].append(max_score)
result["label"] = str(result["label"])
result["prob"] = str(result["prob"])
return result, None, ""
......
......@@ -1343,7 +1343,7 @@ class ResponseOp(Op):
type(var)))
_LOGGER.error("(logid={}) Failed to pack RPC "
"response package: {}".format(
channeldata.id, resp.error_info))
channeldata.id, resp.err_msg))
break
resp.value.append(var)
resp.key.append(name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册