mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix reversed H & W padding for swin patch merging

fix_swin_pad
Ross Wightman 2024-09-21 16:51:02 -07:00
parent ee5b1e8217
commit 65564f7da5
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/swin_transformer.py
View File

@ -435,7 +435,7 @@ class PatchMerging(nn.Module):
def forward(self, x):
B, H, W, C = x.shape
pad_values = (0, 0, 0, H % 2, 0, W % 2)
pad_values = (0, 0, 0, W % 2, 0, H % 2)
x = nn.functional.pad(x, pad_values)
_, H, W, _ = x.shape

2
timm/models/swin_transformer_v2.py
View File

@ -439,7 +439,7 @@ class PatchMerging(nn.Module):
def forward(self, x: torch.Tensor) -> torch.Tensor:
B, H, W, C = x.shape
pad_values = (0, 0, 0, H % 2, 0, W % 2)
pad_values = (0, 0, 0, W % 2, 0, H % 2)
x = nn.functional.pad(x, pad_values)
_, H, W, _ = x.shape

2
timm/models/swin_transformer_v2_cr.py
View File

@ -445,7 +445,7 @@ class PatchMerging(nn.Module):
"""
B, H, W, C = x.shape
pad_values = (0, 0, 0, H % 2, 0, W % 2)
pad_values = (0, 0, 0, W % 2, 0, H % 2)
x = nn.functional.pad(x, pad_values)
_, H, W, _ = x.shape