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

fix imangnet pipeline err

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