Update davit.py

pull/1630/head
Fredo Guan 3 years ago
parent 6d135d22c7
commit 4ba893ceae

@ -458,7 +458,7 @@ class DaViT(nn.Module):
self.apply(self._init_weights)
self.forward = _get_forward_fn()
self.forward = self._get_forward_fn()
'''
if self._features_only == True:
self.forward = self.forward_features_full
@ -480,7 +480,7 @@ class DaViT(nn.Module):
@features_only.setter
def features_only(self, new_value):
self._features_only = new_value
self.forward = _get_forward_fn()
self.forward = self._get_forward_fn()

Loading…
Cancel
Save