Skip to content

Commit b636261

Browse files
committed
fix format and lint checks
1 parent b7b5a37 commit b636261

File tree

1 file changed

+6
-2
lines changed
  • segmentation_models_pytorch/decoders/upernet

1 file changed

+6
-2
lines changed

segmentation_models_pytorch/decoders/upernet/decoder.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,9 @@ def __init__(
3838
def forward(self, x):
3939
_, _, height, weight = x.shape
4040
out = [x] + [
41-
F.interpolate(block(x), size=(height, weight), mode="bilinear", align_corners=False)
41+
F.interpolate(
42+
block(x), size=(height, weight), mode="bilinear", align_corners=False
43+
)
4244
for block in self.blocks
4345
]
4446
out = self.out_conv(torch.cat(out, dim=1))
@@ -61,7 +63,9 @@ def __init__(self, skip_channels, pyramid_channels, use_bathcnorm=True):
6163

6264
def forward(self, x, skip):
6365
_, channels, height, weight = skip.shape
64-
x = F.interpolate(x, size=(height, weight), mode="bilinear", align_corners=False)
66+
x = F.interpolate(
67+
x, size=(height, weight), mode="bilinear", align_corners=False
68+
)
6569
if channels != 0:
6670
skip = self.skip_conv(skip)
6771
x = x + skip

0 commit comments

Comments
 (0)