Update README.md
Browse files
README.md
CHANGED
|
@@ -6,3 +6,20 @@ library_name: timm
|
|
| 6 |
license: apache-2.0
|
| 7 |
---
|
| 8 |
# Model card for eva02_base_patch16_224.mim_in22k
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
license: apache-2.0
|
| 7 |
---
|
| 8 |
# Model card for eva02_base_patch16_224.mim_in22k
|
| 9 |
+
|
| 10 |
+
Downloaded checkpoint from [here](https://huggingface.co/Yuxin-CV/EVA-02/blob/main/eva02/pt/eva02_B_pt_in21k_p14to16.pt)
|
| 11 |
+
|
| 12 |
+
Then did
|
| 13 |
+
```python
|
| 14 |
+
import torch
|
| 15 |
+
import timm
|
| 16 |
+
from timm.models.eva import checkpoint_filter_fn
|
| 17 |
+
from timm.models import push_to_hf_hub
|
| 18 |
+
|
| 19 |
+
state_dict = torch.load("eva02/pt/eva02_B_pt_in21k_p14to16.pt")["model"]
|
| 20 |
+
model = timm.create_model("eva02_base_patch14_224", patch_size=16, ref_feat_shape=(14, 14))
|
| 21 |
+
|
| 22 |
+
filtered = checkpoint_filter_fn(state_dict, model)
|
| 23 |
+
push_to_hf_hub(t, "nomic-ai/eva02_base_patch16_224.mim_in22k", private=True, safe_serialization=True)
|
| 24 |
+
```
|
| 25 |
+
|