Update davit.py

3 years ago · e643c36488
parent 0de8313183
commit e643c36488
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -482,7 +482,7 @@ class DaViT(nn.Module):
        else:
            return self.forward_classification
    '''
-    
+    @torch.jit.ignore
    def _update_forward_fn(self):
        if self._features_only == True:
            self.forward = self.forward_features_full