diff --git a/mace/python/tools/converter_tool/onnx_converter.py b/mace/python/tools/converter_tool/onnx_converter.py index 6befa478318ec9f3fb5644317acf3b3f38ecb6bc..fff5241c9c1372e5e716ae04b3eb494ce7bb95a9 100644 --- a/mace/python/tools/converter_tool/onnx_converter.py +++ b/mace/python/tools/converter_tool/onnx_converter.py @@ -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,