From 98c6934e899e4c42cb1a1ac85a33098e5d7f3c6f Mon Sep 17 00:00:00 2001 From: Wenyu Date: Tue, 19 Oct 2021 10:57:25 +0800 Subject: [PATCH] support export (#4326) --- ppdet/modeling/backbones/swin_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/backbones/swin_transformer.py b/ppdet/modeling/backbones/swin_transformer.py index 8ee8bd66a..73da7958f 100644 --- a/ppdet/modeling/backbones/swin_transformer.py +++ b/ppdet/modeling/backbones/swin_transformer.py @@ -537,7 +537,7 @@ class PatchEmbed(nn.Layer): B, C, H, W = x.shape # assert [H, W] == self.img_size[:2], "Input image size ({H}*{W}) doesn't match model ({}*{}).".format(H, W, self.img_size[0], self.img_size[1]) if W % self.patch_size[1] != 0: - x = F.pad(x, [0, self.patch_size[1] - W % self.patch_size[1]]) + x = F.pad(x, [0, self.patch_size[1] - W % self.patch_size[1], 0, 0]) if H % self.patch_size[0] != 0: x = F.pad(x, [0, 0, 0, self.patch_size[0] - H % self.patch_size[0]]) -- GitLab