From d5eaf309769966d5687b2be1e606c04c7f8dd69d Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Sat, 7 Jan 2023 20:28:40 -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 509fd93f..45d60a67 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -534,7 +534,7 @@ class DaViT(nn.Module):
         self.head = ClassifierHead(self.num_features, num_classes, pool_type=global_pool, drop_rate=self.drop_rate)
 
     def forward_features(self, x):
-        x = self.patch_embed(x)
+        x = self.stem(x)
         x = self.stages(x)
         x = self.norm(x.permute(0, 2, 3, 1)).permute(0, 3, 1, 2)
         return x