mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #2285 from huggingface/fix_swin_pad

Browse Source 
Fix reversed H & W padding for swin patch merging
This commit is contained in:
Ross Wightman 2024-09-21 17:00:49 -07:00 committed by GitHub
commit 93374637c6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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): def forward(self, x):
B, H, W, C = x.shape 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) x = nn.functional.pad(x, pad_values)
_, H, W, _ = x.shape _, 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: def forward(self, x: torch.Tensor) -> torch.Tensor:
B, H, W, C = x.shape 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) x = nn.functional.pad(x, pad_values)
_, H, W, _ = x.shape _, 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 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) x = nn.functional.pad(x, pad_values)
_, H, W, _ = x.shape _, H, W, _ = x.shape