Hugging Face's logo

Spaces:
ThunderVVV
/
HaWoR
Running

App Files Community
2
HaWoR
File size: 729 Bytes 
b7eedf7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 
db_info={}



db_info['checkpoint']={
    'db_root': 'tbd_weight_root', # Config your weight root!

    # pretrained weight for vit
    'vit_small_reg':  'vit/dinov2_vits14_reg4_pretrain.pth',
    'vit_large_reg':  'vit/dinov2_vitl14_reg4_pretrain.pth',
    'vit_giant2_reg':  'vit/dinov2_vitg14_reg4_pretrain.pth',

    'vit_large': 'vit/dinov2_vitl14_pretrain.pth',

    # pretrained weight for convnext
    'convnext_tiny': 'convnext/convnext_tiny_22k_1k_384.pth',
    'convnext_small': 'convnext/convnext_small_22k_1k_384.pth',
    'convnext_base': 'convnext/convnext_base_22k_1k_384.pth',
    'convnext_large': 'convnext/convnext_large_22k_1k_384.pth',
    'convnext_xlarge': 'convnext/convnext_xlarge_22k_1k_384_ema.pth',
}