From bdbc0262a5846dca0c99533d1834e4d2d3678376 Mon Sep 17 00:00:00 2001 From: wjj19950828 Date: Thu, 4 Aug 2022 21:22:04 +0800 Subject: [PATCH] fixed pad bug --- x2paddle/op_mapper/pytorch2paddle/aten.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/pytorch2paddle/aten.py b/x2paddle/op_mapper/pytorch2paddle/aten.py index 8153790..0212f3d 100755 --- a/x2paddle/op_mapper/pytorch2paddle/aten.py +++ b/x2paddle/op_mapper/pytorch2paddle/aten.py @@ -1221,7 +1221,9 @@ def aten_constant_pad_nd(mapper, graph, node): layer_attrs["value"] = mapper.attrs[inputs_name[2]] if padding_attr is not None: - padding_attr.reverse() + ## convert torch pad attr to paddle pad attr, eg:(x1,x2,x3,x4)->(x3,x4,x1,x2) + padding_attr = np.array(padding_attr).reshape((-1, 2)) + padding_attr = np.flip(padding_attr, axis=0).flatten().tolist() layer_inputs["x"] = inputs_name[0] kernel_name = "paddle.nn.functional.pad" if len(padding_attr) == 2: -- GitLab