Update davit.py

pull/1630/head
Fredo Guan 3 years ago
parent 5a9f582271
commit 1d7c04e28a

@ -382,7 +382,7 @@ class DaViT(nn.Module):
attn_drop_rate=0.,
img_size=224,
num_classes=1000,
global_pool='avg'#,
global_pool='avg',
#features_only = False
**kwargs):
super().__init__()

Loading…
Cancel
Save