From d1e68fa33f03ecd29223c268a1fa43e2c361b432 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 00:58:07 -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 fd13161b..a404a4b4 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -478,7 +478,7 @@ class DaViT(nn.Module): return self._features_only @features_only.setter - def features_only(self, new_value): + def features_only(self, new_value : bool): self._features_only = new_value self.forward = self._get_forward_fn()