From a714da42cbbc769c560699e939235b7fa56a23f0 Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Fri, 8 Jan 2021 16:54:22 +0800 Subject: [PATCH] fix --- x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py | 1 + x2paddle/op_mapper/static/onnx2paddle/opset9/opset.py | 1 + 2 files changed, 2 insertions(+) diff --git a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py index 5bab894..24c35cc 100644 --- a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py +++ b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py @@ -302,6 +302,7 @@ class OpSet9(): val_x_shape = val_x.out_shapes[0] if mode == "linear" and len(val_x_shape) == 4: attrs["mode"] = string("bilinear") + attrs["align_corners"] = True self.paddle_graph.add_layer( kernel="paddle.nn.functional.interpolate", inputs=inputs, diff --git a/x2paddle/op_mapper/static/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/static/onnx2paddle/opset9/opset.py index 889424d..1c89882 100644 --- a/x2paddle/op_mapper/static/onnx2paddle/opset9/opset.py +++ b/x2paddle/op_mapper/static/onnx2paddle/opset9/opset.py @@ -292,6 +292,7 @@ class OpSet9(): val_x_shape = val_x.out_shapes[0] if mode == "linear" and len(val_x_shape) == 4: attrs["mode"] = string("bilinear") + attrs["align_corners"] = True self.paddle_graph.add_layer( kernel="paddle.nn.functional.interpolate", inputs=inputs, -- GitLab