zpn's picture
Update README.md
4ecf59a verified
metadata
tags:
  - image-classification
  - timm
library_name: timm
license: apache-2.0

Model card for eva02_base_patch16_224.mim_in22k

Downloaded checkpoint from here

Then did

import torch
import timm
from timm.models.eva import checkpoint_filter_fn
from timm.models import push_to_hf_hub

state_dict = torch.load("eva02/pt/eva02_B_pt_in21k_p14to16.pt")["model"]
model = timm.create_model("eva02_base_patch14_224", patch_size=16, ref_feat_shape=(14, 14))

filtered = checkpoint_filter_fn(state_dict, model)
push_to_hf_hub(t, "nomic-ai/eva02_base_patch16_224.mim_in22k", private=True, safe_serialization=True)