Skip to content

Commit 94195a9

Browse files
bug: Fix PatchMerging duplicate merging
1 parent 9eb0a8c commit 94195a9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

monai/networks/nets/swin_unetr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -783,8 +783,8 @@ def forward(self, x):
783783
x2 = x[:, 0::2, 1::2, 0::2, :]
784784
x3 = x[:, 0::2, 0::2, 1::2, :]
785785
x4 = x[:, 1::2, 0::2, 1::2, :]
786-
x5 = x[:, 0::2, 1::2, 0::2, :]
787-
x6 = x[:, 0::2, 0::2, 1::2, :]
786+
x5 = x[:, 1::2, 1::2, 0::2, :]
787+
x6 = x[:, 0::2, 1::2, 1::2, :]
788788
x7 = x[:, 1::2, 1::2, 1::2, :]
789789
x = torch.cat([x0, x1, x2, x3, x4, x5, x6, x7], -1)
790790
x = self.norm(x)

0 commit comments

Comments
 (0)