未验证 提交 04db6237 编写于 作者: L LiuTuo 提交者: GitHub

Merge pull request #388 from yytdfc/dev_fix_onnx

fix blank name for onnx model, use output name instead
......@@ -230,6 +230,8 @@ def onnx_dtype(dtype):
class OnnxNode(object):
def __init__(self, node):
self.name = str(node.name)
if self.name == '':
self.name = str(node.output)
self.op_type = str(node.op_type)
self.domain = str(node.domain)
self.attrs = dict([(attr.name,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册