From 00c7a7e32fc5de3824a3c7df8380d86cd7011f20 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 02:46:15 -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 61393e59..61eeba9c 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -399,7 +399,7 @@ class DaViT(nn.Module):
         self.num_features = embed_dims[-1]
         self.drop_rate=drop_rate
         self.grad_checkpointing = False
-        self._features_only = kwargs.get(features_only, False)
+        self._features_only = kwargs.get('features_only', False)
         self.feature_info = []
 
         self.patch_embeds = nn.ModuleList([