diff --git a/timm/models/swin_transformer_v2.py b/timm/models/swin_transformer_v2.py index d2adf01c..bffdb1d1 100644 --- a/timm/models/swin_transformer_v2.py +++ b/timm/models/swin_transformer_v2.py @@ -164,7 +164,7 @@ class WindowAttention(nn.Module): torch.meshgrid( [ relative_coords_h, - relative_coords_w + relative_coords_w, ], indexing='ij', )