try modify swin

2024-04-29 17:16:51 +07:00 · 2024-04-29 17:16:51 +07:00 · 8866ea448e
commit 8866ea448e
parent b661823661
1 changed files with 0 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -978,7 +978,6 @@ class PoolFormerBlock(nn.Module):
                    img_mask[:, :, h, w] = cnt
                    cnt += 1

-            print(self.input_resolution)
            mask_windows = window_partition(img_mask, self.window_size)  # nW, window_size, window_size, 1
            mask_windows = mask_windows.view(-1, self.window_size * self.window_size)
            attn_mask = mask_windows.unsqueeze(1) - mask_windows.unsqueeze(2)