fix Swin

gau-nernst · Oct 29, 2023 · cad7d5b · cad7d5b
1 parent ad22e50
commit cad7d5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_toolbox/backbones/swin.py b/vision_toolbox/backbones/swin.py
@@ -78,7 +78,7 @@ def forward(self, x: Tensor) -> Tensor:
             attn_bias = attn_bias + self.attn_mask.unsqueeze(1)  # add n_heads dim
 
         x, nH, nW = window_partition(x, self.window_size)  # (B * nH * nW, win_size * win_size, C)
-        x = super().forward(x, attn_bias)
+        x = super().forward(x, attn_bias=attn_bias)
         x = window_unpartition(x, self.window_size, nH, nW)  # (B, H, W, C)
 
         if self.shift > 0: