From 3a4c89d659e2eaf35a9c22d6d587100abcc4a9ec Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Sat, 10 Dec 2022 04:35:58 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 7a6d8142..fe2c1bf3 100644
--- 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