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 #1918 from Mingosnake/fix-efficientformerv2-typo

Browse Source 
Fix typo in efficientformer_v2
This commit is contained in:
Ross Wightman 2023-08-15 20:04:35 -07:00 committed by GitHub
commit 24fc720d34
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientformer_v2.py
View File

@ -232,7 +232,7 @@ class Attention2dDownsample(torch.nn.Module):
self.attention_biases = nn.Parameter(torch.zeros(num_heads, self.N)) self.attention_biases = nn.Parameter(torch.zeros(num_heads, self.N))
k_pos = torch.stack(torch.meshgrid(torch.arange( k_pos = torch.stack(torch.meshgrid(torch.arange(
self.resolution[1]), self.resolution[0]),
torch.arange(self.resolution[1]))).flatten(1) torch.arange(self.resolution[1]))).flatten(1)
q_pos = torch.stack(torch.meshgrid( q_pos = torch.stack(torch.meshgrid(
torch.arange(0, self.resolution[0], step=2), torch.arange(0, self.resolution[0], step=2),