提交 af8e70e2 编写于 作者: F fuchen

fix blank name for onnx model, use output name instead

上级 2a563185
......@@ -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.
先完成此消息的编辑!
想要评论请 注册