Update davit.py

pull/1630/head
Fredo Guan 3 years ago
parent f8c5150fcd
commit 36527b5a33

@ -572,7 +572,7 @@ class DaViT(nn.Module):
# non-normalized pyramid features + corresponding sizes # non-normalized pyramid features + corresponding sizes
return tuple(features), tuple(sizes) return features, sizes
def forward_features(self, x): def forward_features(self, x):
x, sizes = self.forward_features_full(x) x, sizes = self.forward_features_full(x)

Loading…
Cancel
Save