From cad7d5b705644a6a38dc2d664d9f600372bf733a Mon Sep 17 00:00:00 2001 From: Thien Tran Date: Sun, 29 Oct 2023 14:43:20 +0800 Subject: [PATCH] fix Swin --- vision_toolbox/backbones/swin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vision_toolbox/backbones/swin.py b/vision_toolbox/backbones/swin.py index da0fe35..b373ba7 100644 --- 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: