From 6d135d22c76491785c06a78f046312fbaa708647 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 00:56:45 -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 9d2cf85f..f0fc52da 100644 --- 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()