Skip to content

Commit 6995164

Browse files
author
valhassan
committed
Fix variable naming in Decoder class to improve code clarity by renaming dimensions in forward pass.
1 parent 51595da commit 6995164

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

geo_deep_learning/models/decoders/segformer_mlp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def __init__( # noqa: PLR0913
7777
def forward(self, x: list[torch.Tensor]) -> torch.Tensor:
7878
"""Forward pass."""
7979
c1, c2, c3, c4 = x
80-
n, _, h, w = c4.shape
80+
n, _, _h, _w = c4.shape
8181

8282
_c4 = (
8383
self.linear_c4(c4)

0 commit comments

Comments
 (0)