Update davit.py

3 years ago · 6d135d22c7
parent 889c2ca728
commit 6d135d22c7
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -477,7 +477,7 @@ class DaViT(nn.Module):
    def features_only(self):
        return self._features_only
        
-    @value.setter
+    @features_only.setter
    def features_only(self, new_value):
        self._features_only = new_value
        self.forward = _get_forward_fn()