Views
No views yet
1import torch
2import timm
3
4# create model
5model = timm.create_model(
6 "vit_small_plus_patch16_dinov3.lvd1689m",
7 pretrained=False,
8 num_classes=39,
9)
10
11# load weights
12state_dict = torch.load("model.bin", map_location="cpu")
13model.load_state_dict(state_dict)