From 7938f28542415cd7dc3922822ff9436a1dbb02be Mon Sep 17 00:00:00 2001 From: Minseo Kang Date: Wed, 16 Aug 2023 03:29:01 +0900 Subject: [PATCH] Fix typo in efficientformer_v2 --- timm/models/efficientformer_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/efficientformer_v2.py b/timm/models/efficientformer_v2.py index 9388131e..357b258d 100644 --- a/timm/models/efficientformer_v2.py +++ b/timm/models/efficientformer_v2.py @@ -232,7 +232,7 @@ class Attention2dDownsample(torch.nn.Module): self.attention_biases = nn.Parameter(torch.zeros(num_heads, self.N)) k_pos = torch.stack(torch.meshgrid(torch.arange( - self.resolution[1]), + self.resolution[0]), torch.arange(self.resolution[1]))).flatten(1) q_pos = torch.stack(torch.meshgrid( torch.arange(0, self.resolution[0], step=2),