Hugging Face's logo

WeiChow
/
dptv2_b_vit

PyTorch
Model card Files Community
WeiChow commited on
Commit
b5eb030
•
1 Parent(s): 09ab8f7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -53,7 +53,7 @@ img = Image.open(urlopen(
53
  ))
54
 
55
  model = timm.create_model(
56
- 'vit_large_patch14_dinov2.lvd142m',
57
  pretrained=True,
58
  num_classes=0, # remove classifier nn.Linear
59
  checkpoint_path="pytorch_model.bin"
 
53
  ))
54
 
55
  model = timm.create_model(
56
+ 'vit_base_patch14_dinov2.lvd142m',
57
  pretrained=True,
58
  num_classes=0, # remove classifier nn.Linear
59
  checkpoint_path="pytorch_model.bin"