提交 ba5a3314 编写于 作者: S SunAhong1993

fix the bug

上级 f6af37b4
......@@ -234,11 +234,7 @@ class ONNXGraph(Graph):
"""
generate output_nodes node of ONNX model
"""
# inner_nodes = self.get_inner_nodes()
output_nodes = [value.name for value in self.graph.output]
# for opt_data in output_nodes:
# if opt_data not in inner_nodes:
# self.output_nodes.append(opt_data)
for opt_data in output_nodes:
n = super(ONNXGraph, self).get_node(opt_data)
if n is None:
......@@ -408,10 +404,8 @@ class ONNXDecoder(object):
check_model(onnx_model)
onnx_model = self.optimize_model_skip_op(onnx_model)
# onnx_model = self.optimize_model_strip_initializer(onnx_model)
onnx_model = self.optimize_node_name(onnx_model)
self.graph = ONNXGraph(onnx_model)
#self.onnx_model = onnx_model
def build_value_refs(self, nodes):
"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册