Update davit.py

3 years ago · 7997ddc30d
parent d35e98b7fa
commit 7997ddc30d
1 changed files with 3 additions and 3 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -458,13 +458,13 @@ class DaViT(nn.Module):
        self.apply(self._init_weights)
-        self.forward = self._get_forward_fn()
+        #self.forward = self._get_forward_fn()
-        '''
+        
        if self._features_only == True:
            self.forward = self.forward_features_full
        else:
            self.forward = self.forward_classification
-        '''
+        
    '''
    def _get_forward_fn(self):