From f19863913ee1b7300e01e973dd55f06e9b050a08 Mon Sep 17 00:00:00 2001 From: shiyutang <34859558+shiyutang@users.noreply.github.com> Date: Thu, 30 Dec 2021 20:10:13 +0800 Subject: [PATCH] [Enhancement] fix encnet export problem --- paddleseg/models/encnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddleseg/models/encnet.py b/paddleseg/models/encnet.py index 19698f1b31..0f909addc5 100644 --- a/paddleseg/models/encnet.py +++ b/paddleseg/models/encnet.py @@ -107,9 +107,9 @@ def forward(self, inputs): if self.add_lateral: laterals = [] - for i, lateral_conv in enumerate(self.lateral_convs): + for j, lateral_conv in enumerate(self.lateral_convs): laterals.append( - F.interpolate(lateral_conv(feats[i]), + F.interpolate(lateral_conv(feats[j]), size=paddle.shape(feat)[2:], mode='bilinear', align_corners=False))