Update davit.py

timm/models/davit.py

@@ -215,7 +215,8 @@ def window_reverse(windows : Tensor, window_size: int, H: int, W: int):
     Returns:
         x: (B, H, W, C)
     """
-    B = int(windows.size(dim=0)[0] / (H * W / window_size / window_size))
+    window_size_dim_0 : int = windows.size(dim=0)
+    B = int(window_size_dim_0 / (H * W / window_size / window_size))
     x = windows.view(B, H // window_size, W // window_size, window_size, window_size, -1)
     x = x.permute(0, 1, 3, 2, 4, 5).contiguous().view(B, H, W, -1)
     return x