From 007be8319ba8b89324bfc308e873be3b24356619 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 00:39:15 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 06f65020..d607f945 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -577,9 +577,9 @@ class DaViT(nn.Module):
     
     def forward(self, x):
         if self.features_only == True:
-            return forward_features_full(x)
+            return self.forward_features_full(x)
         else:
-            return forward(x)
+            return self.forward_classification(x)
 
 def checkpoint_filter_fn(state_dict, model):
     """ Remap MSFT checkpoints -> timm """