From ba5a33141f78054a95e76283ab1c8feceb5245ec Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Thu, 7 Jan 2021 14:20:53 +0800 Subject: [PATCH] fix the bug --- x2paddle/decoder/onnx_decoder.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/x2paddle/decoder/onnx_decoder.py b/x2paddle/decoder/onnx_decoder.py index 3db83a8..dac697d 100644 --- a/x2paddle/decoder/onnx_decoder.py +++ b/x2paddle/decoder/onnx_decoder.py @@ -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): """ -- GitLab