Hugging Face's logo

edadaltocg
/
vit_base_patch16_224_in21k_ft_cifar100

Image Classification
timm
PyTorch
English
timm/vit_base_patch16_224.orig_in21k_ft_in1k
cifar100
Eval Results
Model card Files Community
vit_base_patch16_224_in21k_ft_cifar100
File size: 1,822 Bytes 
c1428de
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6c8baa3
 
 
 
 
5cc7bb8
6c8baa3
 
 
 
 
 
 
c1428de
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
 
---
language: en
license: mit
library_name: timm
tags:
- image-classification
- timm/vit_base_patch16_224.orig_in21k_ft_in1k
- cifar100
datasets: cifar100
metrics:
- accuracy
model-index:
- name: vit_base_patch16_224_in21k_ft_cifar100
  results:
  - task:
      type: image-classification
    dataset:
      name: CIFAR-100
      type: cifar100
    metrics:
    - type: accuracy
      value: 0.9316
---

# Model Card for Model ID

This model is a small timm/vit_base_patch16_224.orig_in21k_ft_in1k trained on cifar100.

- **Test Accuracy:** 0.9316
- **License:** MIT

## How to Get Started with the Model

Use the code below to get started with the model.

```python
import timm
import torch
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, 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",
        map_location="cpu",
        file_name="vit_base_patch16_224_in21k_ft_cifar100.pth",
    )
)
```

## Training Data

Training data is cifar100.

## Training Hyperparameters


- **config**: `scripts/train_configs/ft_cifar100.json`

- **model**: `vit_base_patch16_224_in21k_ft_cifar100`

- **dataset**: `cifar100`

- **batch_size**: `64`

- **epochs**: `10`

- **validation_frequency**: `1`

- **seed**: `1`

- **criterion**: `CrossEntropyLoss`

- **criterion_kwargs**: `{}`

- **optimizer**: `SGD`

- **lr**: `0.01`

- **optimizer_kwargs**: `{'momentum': 0.9, 'weight_decay': 0.0}`

- **scheduler**: `CosineAnnealingLR`

- **scheduler_kwargs**: `{'T_max': 10}`

- **debug**: `False`


## Testing Data

Testing data is cifar100.

---

This model card was created by Eduardo Dadalto.