From 2faeaebfee765452cfac9c2446471ccc1606ef27 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 01:58:34 -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 d7e29fe5..15429003 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -607,7 +607,7 @@ class DaViT(nn.Module):
         outs = []
         for i, out in enumerate(x):
             H, W = sizes[i]
-            outs.append(x.view(-1, H, W, self.embed_dims[i]).permute(0, 3, 1, 2).contiguous())
+            outs.append(out.view(-1, H, W, self.embed_dims[i]).permute(0, 3, 1, 2).contiguous())
         
         return outs