edadaltocg
/

vit_base_patch16_224_in21k_ft_cifar100

Image Classification

timm/vit_base_patch16_224.orig_in21k_ft_in1k

Model card Files Files and versions Community

edadaltocg commited on Mar 3, 2023

Commit

5cc7bb8

•

1 Parent(s): 6c8baa3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ from torch import nn
 model = timm.create_model("timm/vit_base_patch16_224.orig_in21k_ft_in1k",
 pretrained=False)
-model.head = nn.Linear(model.head.in_features, 10)
 model.load_state_dict(
     torch.hub.load_state_dict_from_url(
         "https://huggingface.co/edadaltocg/vit_base_patch16_224_in21k_ft_cifar100/resolve/main/pytorch_model.bin",

 model = timm.create_model("timm/vit_base_patch16_224.orig_in21k_ft_in1k",
 pretrained=False)
+model.head = nn.Linear(model.head.in_features, 100)
 model.load_state_dict(
     torch.hub.load_state_dict_from_url(
         "https://huggingface.co/edadaltocg/vit_base_patch16_224_in21k_ft_cifar100/resolve/main/pytorch_model.bin",