Skip to content

Commit 18921b2

Browse files
committed
Compatibility with newer version of xception code
1 parent 71a6737 commit 18921b2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

segmentation_models_pytorch/encoders/xception.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,10 @@ def forward(self, x):
4242
if self._depth >= 1:
4343
x = self.conv1(x)
4444
x = self.bn1(x)
45-
x = self.relu(x)
45+
x = self.relu1(x)
4646
x = self.conv2(x)
4747
x = self.bn2(x)
48-
x = self.relu(x)
48+
x = self.relu2(x)
4949
features.append(x)
5050

5151
if self._depth >= 2:
@@ -72,7 +72,7 @@ def forward(self, x):
7272
x = self.block12(x)
7373
x = self.conv3(x)
7474
x = self.bn3(x)
75-
x = self.relu(x)
75+
x = self.relu3(x)
7676
x = self.conv4(x)
7777
x = self.bn4(x)
7878
features.append(x)

0 commit comments

Comments
 (0)