edadaltocg commited on
Commit
5cc7bb8
1 Parent(s): 6c8baa3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ from torch import nn
40
 
41
  model = timm.create_model("timm/vit_base_patch16_224.orig_in21k_ft_in1k",
42
  pretrained=False)
43
- model.head = nn.Linear(model.head.in_features, 10)
44
  model.load_state_dict(
45
  torch.hub.load_state_dict_from_url(
46
  "https://huggingface.co/edadaltocg/vit_base_patch16_224_in21k_ft_cifar100/resolve/main/pytorch_model.bin",
 
40
 
41
  model = timm.create_model("timm/vit_base_patch16_224.orig_in21k_ft_in1k",
42
  pretrained=False)
43
+ model.head = nn.Linear(model.head.in_features, 100)
44
  model.load_state_dict(
45
  torch.hub.load_state_dict_from_url(
46
  "https://huggingface.co/edadaltocg/vit_base_patch16_224_in21k_ft_cifar100/resolve/main/pytorch_model.bin",