rename variable

Author: brianhou0208

segmentation_models_pytorch/decoders/upernet/decoder.py

@@ -36,9 +36,9 @@ def __init__(
         )
 
     def forward(self, x):
-        _, _, h, w = x.shape
+        _, _, height, weight = x.shape
         out = [x] + [
-            F.interpolate(block(x), size=(h, w), mode="bilinear", align_corners=False)
+            F.interpolate(block(x), size=(height, weight), mode="bilinear", align_corners=False)
             for block in self.blocks
         ]
         out = self.out_conv(torch.cat(out, dim=1))
@@ -60,9 +60,9 @@ def __init__(self, skip_channels, pyramid_channels, use_bathcnorm=True):
         )
 
     def forward(self, x, skip):
-        _, ch, h, w = skip.shape
-        x = F.interpolate(x, size=(h, w), mode="bilinear", align_corners=False)
-        if ch != 0:
+        _, channels, height, weight = skip.shape
+        x = F.interpolate(x, size=(height, weight), mode="bilinear", align_corners=False)
+        if channels != 0:
             skip = self.skip_conv(skip)
             x = x + skip
         return x