fix xception flow to enable model export (#400)

834cc164 · littletomatodonkey · GitHub · 10e73342 · 834cc164
隐藏空白更改
内联并排

Showing with 7 addition and 6 deletion

ppcls/modeling/architectures/xception_deeplab.py ppcls/modeling/architectures/xception_deeplab.py +7 -6

未找到文件。
--- a/ppcls/modeling/architectures/xception_deeplab.py
+++ b/ppcls/modeling/architectures/xception_deeplab.py
@@ -238,13 +238,14 @@ class Xception_Block(nn.Layer):
            x = self._conv1(inputs)
            x = self._conv2(x)
            x = self._conv3(x)
-        if self.has_skip is False:
+        if self.has_skip:
-            return x
+            if self.skip_conv:
-        if self.skip_conv:
+                skip = self._short(inputs)
-            skip = self._short(inputs)
+            else:
+                skip = inputs
+            return paddle.add(x, skip)
        else:
-            skip = inputs
+            return x
-        return paddle.add(x, skip)
 class XceptionDeeplab(nn.Layer):