diff --git a/x2paddle/decoder/onnx_decoder.py b/x2paddle/decoder/onnx_decoder.py index 3db83a895fd6198f7f08f36d10f033f472b806d9..dac697d4d28a6e00363879307e0db328e056d63e 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): """