未验证 提交 834cc164 编写于 作者: L littletomatodonkey 提交者: GitHub

fix xception flow to enable model export (#400)

上级 10e73342
......@@ -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:
return x
if self.skip_conv:
skip = self._short(inputs)
if self.has_skip:
if self.skip_conv:
skip = self._short(inputs)
else:
skip = inputs
return paddle.add(x, skip)
else:
skip = inputs
return paddle.add(x, skip)
return x
class XceptionDeeplab(nn.Layer):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册