Hugging Face's logo

davanstrien
/
vit-base-patch32-224-in21-leicester_binary

Image Classification
Transformers
PyTorch
TensorBoard
vit
vision
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
2
davanstrien HF staff commited on
Commit
3958293
·
1 Parent(s): 6a77138

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +43 -45
README.md CHANGED
@@ -1,8 +1,6 @@
1
  ---
2
  license: apache-2.0
3
  tags:
4
- - image-classification
5
- - vision
6
  - generated_from_trainer
7
  metrics:
8
  - f1
@@ -16,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # vit-base-patch32-224-in21-leicester_binary
18
 
19
- This model is a fine-tuned version of [google/vit-base-patch32-224-in21k](https://huggingface.co/google/vit-base-patch32-224-in21k) on the davanstrien/leicester_loaded_annotations_binary dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0949
22
- - F1: 0.9747
23
 
24
  ## Model description
25
 
@@ -51,46 +49,46 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | F1 |
53
  |:-------------:|:-----:|:----:|:---------------:|:------:|
54
- | No log | 1.0 | 7 | 0.4576 | 0.8608 |
55
- | 0.5021 | 2.0 | 14 | 0.3953 | 0.8608 |
56
- | 0.3595 | 3.0 | 21 | 0.3809 | 0.8608 |
57
- | 0.3595 | 4.0 | 28 | 0.3286 | 0.8608 |
58
- | 0.3009 | 5.0 | 35 | 0.2945 | 0.8608 |
59
- | 0.2843 | 6.0 | 42 | 0.3528 | 0.8608 |
60
- | 0.2843 | 7.0 | 49 | 0.2345 | 0.8608 |
61
- | 0.266 | 8.0 | 56 | 0.2499 | 0.8608 |
62
- | 0.222 | 9.0 | 63 | 0.2544 | 0.8608 |
63
- | 0.2018 | 10.0 | 70 | 0.1954 | 0.8608 |
64
- | 0.2018 | 11.0 | 77 | 0.2351 | 0.8608 |
65
- | 0.1948 | 12.0 | 84 | 0.1705 | 0.8608 |
66
- | 0.2053 | 13.0 | 91 | 0.1625 | 0.8734 |
67
- | 0.2053 | 14.0 | 98 | 0.1719 | 0.9367 |
68
- | 0.1729 | 15.0 | 105 | 0.1489 | 0.9367 |
69
- | 0.1535 | 16.0 | 112 | 0.1450 | 0.9494 |
70
- | 0.1535 | 17.0 | 119 | 0.1750 | 0.9494 |
71
- | 0.1492 | 18.0 | 126 | 0.1514 | 0.9494 |
72
- | 0.1349 | 19.0 | 133 | 0.1304 | 0.9620 |
73
- | 0.1538 | 20.0 | 140 | 0.1291 | 0.9620 |
74
- | 0.1538 | 21.0 | 147 | 0.1306 | 0.9620 |
75
- | 0.1357 | 22.0 | 154 | 0.1283 | 0.9620 |
76
- | 0.147 | 23.0 | 161 | 0.1289 | 0.9494 |
77
- | 0.147 | 24.0 | 168 | 0.1339 | 0.9747 |
78
- | 0.1388 | 25.0 | 175 | 0.1244 | 0.9494 |
79
- | 0.1192 | 26.0 | 182 | 0.1117 | 0.9747 |
80
- | 0.1192 | 27.0 | 189 | 0.1105 | 0.9873 |
81
- | 0.112 | 28.0 | 196 | 0.1079 | 0.9747 |
82
- | 0.1215 | 29.0 | 203 | 0.1151 | 0.9620 |
83
- | 0.1139 | 30.0 | 210 | 0.1008 | 0.9873 |
84
- | 0.1139 | 31.0 | 217 | 0.1033 | 0.9747 |
85
- | 0.1164 | 32.0 | 224 | 0.0985 | 0.9873 |
86
- | 0.1192 | 33.0 | 231 | 0.0955 | 0.9873 |
87
- | 0.1192 | 34.0 | 238 | 0.1077 | 0.9620 |
88
- | 0.1132 | 35.0 | 245 | 0.1107 | 0.9620 |
89
- | 0.1021 | 36.0 | 252 | 0.0958 | 0.9873 |
90
- | 0.1021 | 37.0 | 259 | 0.0957 | 0.9873 |
91
- | 0.0945 | 38.0 | 266 | 0.0951 | 0.9747 |
92
- | 0.1244 | 39.0 | 273 | 0.0949 | 0.9747 |
93
- | 0.1012 | 40.0 | 280 | 0.0955 | 0.9873 |
94
 
95
 
96
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
  tags:
 
 
4
  - generated_from_trainer
5
  metrics:
6
  - f1
 
14
 
15
  # vit-base-patch32-224-in21-leicester_binary
16
 
17
+ This model is a fine-tuned version of [google/vit-base-patch32-224-in21k](https://huggingface.co/google/vit-base-patch32-224-in21k) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.1499
20
+ - F1: 0.9494
21
 
22
  ## Model description
23
 
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | F1 |
51
  |:-------------:|:-----:|:----:|:---------------:|:------:|
52
+ | No log | 1.0 | 7 | 0.4529 | 0.8608 |
53
+ | 0.5024 | 2.0 | 14 | 0.3989 | 0.8608 |
54
+ | 0.3533 | 3.0 | 21 | 0.3741 | 0.8608 |
55
+ | 0.3533 | 4.0 | 28 | 0.3161 | 0.8608 |
56
+ | 0.285 | 5.0 | 35 | 0.2824 | 0.8608 |
57
+ | 0.2491 | 6.0 | 42 | 0.2701 | 0.8608 |
58
+ | 0.2491 | 7.0 | 49 | 0.2062 | 0.9114 |
59
+ | 0.2032 | 8.0 | 56 | 0.2050 | 0.9494 |
60
+ | 0.157 | 9.0 | 63 | 0.2013 | 0.9494 |
61
+ | 0.1127 | 10.0 | 70 | 0.1960 | 0.9367 |
62
+ | 0.1127 | 11.0 | 77 | 0.1417 | 0.9494 |
63
+ | 0.0903 | 12.0 | 84 | 0.1307 | 0.9494 |
64
+ | 0.0922 | 13.0 | 91 | 0.0870 | 0.9873 |
65
+ | 0.0922 | 14.0 | 98 | 0.2048 | 0.9241 |
66
+ | 0.0595 | 15.0 | 105 | 0.1204 | 0.9620 |
67
+ | 0.0527 | 16.0 | 112 | 0.2553 | 0.9367 |
68
+ | 0.0527 | 17.0 | 119 | 0.1675 | 0.9367 |
69
+ | 0.0477 | 18.0 | 126 | 0.2265 | 0.9241 |
70
+ | 0.0411 | 19.0 | 133 | 0.1901 | 0.9367 |
71
+ | 0.0299 | 20.0 | 140 | 0.2423 | 0.9241 |
72
+ | 0.0299 | 21.0 | 147 | 0.0639 | 0.9873 |
73
+ | 0.0487 | 22.0 | 154 | 0.1255 | 0.9494 |
74
+ | 0.0359 | 23.0 | 161 | 0.1213 | 0.9494 |
75
+ | 0.0359 | 24.0 | 168 | 0.0727 | 0.9747 |
76
+ | 0.0302 | 25.0 | 175 | 0.1116 | 0.9494 |
77
+ | 0.0304 | 26.0 | 182 | 0.1062 | 0.9494 |
78
+ | 0.0304 | 27.0 | 189 | 0.2097 | 0.9241 |
79
+ | 0.0274 | 28.0 | 196 | 0.1276 | 0.9494 |
80
+ | 0.0291 | 29.0 | 203 | 0.0967 | 0.9494 |
81
+ | 0.0202 | 30.0 | 210 | 0.0765 | 0.9747 |
82
+ | 0.0202 | 31.0 | 217 | 0.0628 | 0.9873 |
83
+ | 0.0232 | 32.0 | 224 | 0.1388 | 0.9494 |
84
+ | 0.0264 | 33.0 | 231 | 0.1062 | 0.9494 |
85
+ | 0.0264 | 34.0 | 238 | 0.1320 | 0.9494 |
86
+ | 0.0219 | 35.0 | 245 | 0.1528 | 0.9494 |
87
+ | 0.0194 | 36.0 | 252 | 0.1746 | 0.9494 |
88
+ | 0.0194 | 37.0 | 259 | 0.1609 | 0.9494 |
89
+ | 0.0204 | 38.0 | 266 | 0.1482 | 0.9494 |
90
+ | 0.0217 | 39.0 | 273 | 0.1522 | 0.9494 |
91
+ | 0.0216 | 40.0 | 280 | 0.1499 | 0.9494 |
92
 
93
 
94
  ### Framework versions