WeiChow commited on
Commit
0e10429
·
verified ·
1 Parent(s): 150b2c2

Upload 2 files

Browse files
Files changed (2) hide show
  1. README.md +19 -3
  2. config.json +34 -0
README.md CHANGED
@@ -1,3 +1,19 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ same architecture with [timm/vit_large_patch14_dinov2.lvd142m](https://huggingface.co/timm/vit_large_patch14_dinov2.lvd142m)
2
+
3
+ ```python
4
+ import torch
5
+ from moge.model import MoGeModel
6
+
7
+ device = torch.device("cuda")
8
+ model = MoGeModel.from_pretrained("Ruicheng/moge-vitl").to(device)
9
+
10
+ # -------------------
11
+ backbone_state_dict = model.backbone.state_dict()
12
+ filtered_state_dict = {k: v for k, v in backbone_state_dict.items() if 'mask_token' not in k}
13
+ torch.save(filtered_state_dict, "moge_l_vit.bin")
14
+ # -------------------
15
+ ```
16
+
17
+
18
+
19
+ Copyright saved.
config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "vit_large_patch14_dinov2",
3
+ "num_classes": 0,
4
+ "num_features": 1024,
5
+ "global_pool": "token",
6
+ "pretrained_cfg": {
7
+ "tag": "lvd142m",
8
+ "custom_load": false,
9
+ "input_size": [
10
+ 3,
11
+ 518,
12
+ 518
13
+ ],
14
+ "fixed_input_size": true,
15
+ "interpolation": "bicubic",
16
+ "crop_pct": 1.0,
17
+ "crop_mode": "center",
18
+ "mean": [
19
+ 0.485,
20
+ 0.456,
21
+ 0.406
22
+ ],
23
+ "std": [
24
+ 0.229,
25
+ 0.224,
26
+ 0.225
27
+ ],
28
+ "num_classes": 0,
29
+ "pool_size": null,
30
+ "first_conv": "patch_embed.proj",
31
+ "classifier": "head",
32
+ "license": "cc-by-nc-4.0"
33
+ }
34
+ }