Update davit.py

3 years ago · 3a4c89d659
parent d5b74ae34f
commit 3a4c89d659
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -582,7 +582,7 @@ class DaViT(nn.Module):
        
        # take final feature and norm
        x = self.norms(x)
-        H, W = sizes
+        H, W = size
        x = x.view(-1, H, W, self.embed_dims[-1]).permute(0, 3, 1, 2).contiguous()
        return x