Update davit.py

3 years ago · 00c7a7e32f
parent 6812acbb72
commit 00c7a7e32f
1 changed files with 1 additions and 1 deletions
--- 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([