Hugging Face's logo

LamaAldakhil
/
SL-CvT

Image Classification
Transformers
PyTorch
TensorBoard
cvt
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
LamaAldakhil commited on
Commit
ecaa061
•
1 Parent(s): a71bf61

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +182 -0
README.md ADDED
@@ -0,0 +1,182 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - imagefolder
7
+ metrics:
8
+ - f1
9
+ - accuracy
10
+ model-index:
11
+ - name: SL-CvT
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: imagefolder
18
+ type: imagefolder
19
+ config: default
20
+ split: train
21
+ args: default
22
+ metrics:
23
+ - name: F1
24
+ type: f1
25
+ value: 0.9269190721647167
26
+ - name: Accuracy
27
+ type: accuracy
28
+ value: 0.9285341679707877
29
+ ---
30
+
31
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
32
+ should probably proofread and complete it, then remove this comment. -->
33
+
34
+ # SL-CvT
35
+
36
+ This model is a fine-tuned version of [microsoft/cvt-13](https://huggingface.co/microsoft/cvt-13) on the imagefolder dataset.
37
+ It achieves the following results on the evaluation set:
38
+ - Loss: 0.3482
39
+ - F1: 0.9269
40
+ - Roc Auc: 0.9789
41
+ - Accuracy: 0.9285
42
+
43
+ ## Model description
44
+
45
+ More information needed
46
+
47
+ ## Intended uses & limitations
48
+
49
+ More information needed
50
+
51
+ ## Training and evaluation data
52
+
53
+ More information needed
54
+
55
+ ## Training procedure
56
+
57
+ ### Training hyperparameters
58
+
59
+ The following hyperparameters were used during training:
60
+ - learning_rate: 5e-05
61
+ - train_batch_size: 32
62
+ - eval_batch_size: 32
63
+ - seed: 42
64
+ - gradient_accumulation_steps: 4
65
+ - total_train_batch_size: 128
66
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
67
+ - lr_scheduler_type: linear
68
+ - lr_scheduler_warmup_ratio: 0.1
69
+ - num_epochs: 100
70
+
71
+ ### Training results
72
+
73
+ | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy |
74
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|
75
+ | 1.2379 | 1.0 | 60 | 1.0716 | 0.6422 | 0.7323 | 0.7246 |
76
+ | 1.0186 | 2.0 | 120 | 0.8477 | 0.6425 | 0.7879 | 0.7293 |
77
+ | 0.9433 | 3.0 | 180 | 0.7473 | 0.7060 | 0.8454 | 0.7538 |
78
+ | 0.8644 | 4.0 | 240 | 0.6831 | 0.7188 | 0.8696 | 0.7663 |
79
+ | 0.7985 | 5.0 | 300 | 0.6420 | 0.7409 | 0.8943 | 0.7799 |
80
+ | 0.7322 | 6.0 | 360 | 0.5713 | 0.7886 | 0.9196 | 0.8101 |
81
+ | 0.725 | 7.0 | 420 | 0.5311 | 0.7989 | 0.9324 | 0.8190 |
82
+ | 0.6529 | 8.0 | 480 | 0.5246 | 0.7852 | 0.9404 | 0.8117 |
83
+ | 0.6224 | 9.0 | 540 | 0.4598 | 0.8282 | 0.9517 | 0.8440 |
84
+ | 0.6315 | 10.0 | 600 | 0.4363 | 0.8457 | 0.9585 | 0.8529 |
85
+ | 0.5651 | 11.0 | 660 | 0.4437 | 0.8323 | 0.9564 | 0.8503 |
86
+ | 0.574 | 12.0 | 720 | 0.4003 | 0.8531 | 0.9617 | 0.8638 |
87
+ | 0.5269 | 13.0 | 780 | 0.3901 | 0.8676 | 0.9671 | 0.8722 |
88
+ | 0.5138 | 14.0 | 840 | 0.3984 | 0.8607 | 0.9685 | 0.8732 |
89
+ | 0.4839 | 15.0 | 900 | 0.3763 | 0.8683 | 0.9701 | 0.8769 |
90
+ | 0.463 | 16.0 | 960 | 0.3398 | 0.8837 | 0.9718 | 0.8894 |
91
+ | 0.4767 | 17.0 | 1020 | 0.3293 | 0.8846 | 0.9738 | 0.8915 |
92
+ | 0.4985 | 18.0 | 1080 | 0.3350 | 0.8852 | 0.9763 | 0.8863 |
93
+ | 0.4657 | 19.0 | 1140 | 0.3369 | 0.8872 | 0.9746 | 0.8951 |
94
+ | 0.4514 | 20.0 | 1200 | 0.3213 | 0.8880 | 0.9750 | 0.8925 |
95
+ | 0.4207 | 21.0 | 1260 | 0.3175 | 0.8943 | 0.9771 | 0.8978 |
96
+ | 0.4522 | 22.0 | 1320 | 0.3229 | 0.8970 | 0.9767 | 0.8983 |
97
+ | 0.4328 | 23.0 | 1380 | 0.3121 | 0.8948 | 0.9791 | 0.8978 |
98
+ | 0.3942 | 24.0 | 1440 | 0.3111 | 0.8993 | 0.9765 | 0.9030 |
99
+ | 0.4414 | 25.0 | 1500 | 0.3062 | 0.9032 | 0.9763 | 0.9061 |
100
+ | 0.3608 | 26.0 | 1560 | 0.3099 | 0.8997 | 0.9787 | 0.9014 |
101
+ | 0.3729 | 27.0 | 1620 | 0.3050 | 0.9029 | 0.9783 | 0.9082 |
102
+ | 0.393 | 28.0 | 1680 | 0.2970 | 0.9090 | 0.9797 | 0.9108 |
103
+ | 0.402 | 29.0 | 1740 | 0.2986 | 0.9087 | 0.9793 | 0.9113 |
104
+ | 0.3697 | 30.0 | 1800 | 0.3384 | 0.8968 | 0.9769 | 0.9025 |
105
+ | 0.3502 | 31.0 | 1860 | 0.3035 | 0.9058 | 0.9789 | 0.9103 |
106
+ | 0.3653 | 32.0 | 1920 | 0.3127 | 0.9024 | 0.9788 | 0.9025 |
107
+ | 0.3898 | 33.0 | 1980 | 0.3222 | 0.9050 | 0.9778 | 0.9061 |
108
+ | 0.317 | 34.0 | 2040 | 0.3013 | 0.9124 | 0.9798 | 0.9139 |
109
+ | 0.3166 | 35.0 | 2100 | 0.3185 | 0.9095 | 0.9775 | 0.9134 |
110
+ | 0.3771 | 36.0 | 2160 | 0.3067 | 0.9049 | 0.9782 | 0.9066 |
111
+ | 0.3487 | 37.0 | 2220 | 0.2948 | 0.9118 | 0.9801 | 0.9134 |
112
+ | 0.3202 | 38.0 | 2280 | 0.2916 | 0.9168 | 0.9788 | 0.9186 |
113
+ | 0.3163 | 39.0 | 2340 | 0.3149 | 0.9141 | 0.9777 | 0.9155 |
114
+ | 0.3605 | 40.0 | 2400 | 0.2964 | 0.9192 | 0.9797 | 0.9207 |
115
+ | 0.3636 | 41.0 | 2460 | 0.3142 | 0.9111 | 0.9810 | 0.9134 |
116
+ | 0.3454 | 42.0 | 2520 | 0.3133 | 0.9111 | 0.9792 | 0.9113 |
117
+ | 0.3561 | 43.0 | 2580 | 0.3090 | 0.9073 | 0.9804 | 0.9077 |
118
+ | 0.3136 | 44.0 | 2640 | 0.3236 | 0.9144 | 0.9782 | 0.9176 |
119
+ | 0.3529 | 45.0 | 2700 | 0.3054 | 0.9175 | 0.9800 | 0.9202 |
120
+ | 0.2987 | 46.0 | 2760 | 0.2944 | 0.9222 | 0.9802 | 0.9233 |
121
+ | 0.2966 | 47.0 | 2820 | 0.3215 | 0.9201 | 0.9786 | 0.9233 |
122
+ | 0.3203 | 48.0 | 2880 | 0.3150 | 0.9219 | 0.9797 | 0.9244 |
123
+ | 0.2821 | 49.0 | 2940 | 0.3072 | 0.9273 | 0.9800 | 0.9291 |
124
+ | 0.2852 | 50.0 | 3000 | 0.3265 | 0.9155 | 0.9792 | 0.9176 |
125
+ | 0.3544 | 51.0 | 3060 | 0.3175 | 0.9150 | 0.9802 | 0.9150 |
126
+ | 0.3327 | 52.0 | 3120 | 0.3134 | 0.9222 | 0.9802 | 0.9244 |
127
+ | 0.2877 | 53.0 | 3180 | 0.3222 | 0.9154 | 0.9805 | 0.9165 |
128
+ | 0.3089 | 54.0 | 3240 | 0.3045 | 0.9248 | 0.9811 | 0.9259 |
129
+ | 0.2904 | 55.0 | 3300 | 0.3301 | 0.9175 | 0.9787 | 0.9186 |
130
+ | 0.2821 | 56.0 | 3360 | 0.3069 | 0.9206 | 0.9810 | 0.9218 |
131
+ | 0.321 | 57.0 | 3420 | 0.3209 | 0.9254 | 0.9800 | 0.9270 |
132
+ | 0.2995 | 58.0 | 3480 | 0.3281 | 0.9202 | 0.9802 | 0.9233 |
133
+ | 0.2683 | 59.0 | 3540 | 0.3263 | 0.9174 | 0.9802 | 0.9202 |
134
+ | 0.3021 | 60.0 | 3600 | 0.3484 | 0.9170 | 0.9788 | 0.9186 |
135
+ | 0.3262 | 61.0 | 3660 | 0.3270 | 0.9151 | 0.9807 | 0.9165 |
136
+ | 0.2329 | 62.0 | 3720 | 0.3280 | 0.9211 | 0.9807 | 0.9233 |
137
+ | 0.2935 | 63.0 | 3780 | 0.3296 | 0.9244 | 0.9807 | 0.9264 |
138
+ | 0.2856 | 64.0 | 3840 | 0.3323 | 0.9209 | 0.9811 | 0.9218 |
139
+ | 0.2829 | 65.0 | 3900 | 0.3390 | 0.9200 | 0.9802 | 0.9218 |
140
+ | 0.3044 | 66.0 | 3960 | 0.3324 | 0.9215 | 0.9799 | 0.9228 |
141
+ | 0.2767 | 67.0 | 4020 | 0.3496 | 0.9150 | 0.9778 | 0.9160 |
142
+ | 0.2936 | 68.0 | 4080 | 0.3378 | 0.9257 | 0.9790 | 0.9275 |
143
+ | 0.2884 | 69.0 | 4140 | 0.3493 | 0.9227 | 0.9790 | 0.9249 |
144
+ | 0.2906 | 70.0 | 4200 | 0.3408 | 0.9259 | 0.9794 | 0.9275 |
145
+ | 0.2542 | 71.0 | 4260 | 0.3559 | 0.9233 | 0.9769 | 0.9249 |
146
+ | 0.2557 | 72.0 | 4320 | 0.3481 | 0.9237 | 0.9779 | 0.9254 |
147
+ | 0.2266 | 73.0 | 4380 | 0.3518 | 0.9208 | 0.9781 | 0.9223 |
148
+ | 0.2771 | 74.0 | 4440 | 0.3544 | 0.9231 | 0.9776 | 0.9254 |
149
+ | 0.2747 | 75.0 | 4500 | 0.3469 | 0.9270 | 0.9780 | 0.9285 |
150
+ | 0.2443 | 76.0 | 4560 | 0.3513 | 0.9216 | 0.9767 | 0.9233 |
151
+ | 0.2859 | 77.0 | 4620 | 0.3456 | 0.9234 | 0.9771 | 0.9254 |
152
+ | 0.2677 | 78.0 | 4680 | 0.3474 | 0.9239 | 0.9780 | 0.9254 |
153
+ | 0.2492 | 79.0 | 4740 | 0.3513 | 0.9235 | 0.9778 | 0.9254 |
154
+ | 0.2532 | 80.0 | 4800 | 0.3524 | 0.9210 | 0.9773 | 0.9233 |
155
+ | 0.2646 | 81.0 | 4860 | 0.3529 | 0.9240 | 0.9784 | 0.9238 |
156
+ | 0.2842 | 82.0 | 4920 | 0.3433 | 0.9260 | 0.9777 | 0.9280 |
157
+ | 0.2872 | 83.0 | 4980 | 0.3584 | 0.9272 | 0.9771 | 0.9285 |
158
+ | 0.2678 | 84.0 | 5040 | 0.3430 | 0.9298 | 0.9777 | 0.9317 |
159
+ | 0.2705 | 85.0 | 5100 | 0.3534 | 0.9268 | 0.9777 | 0.9291 |
160
+ | 0.2605 | 86.0 | 5160 | 0.3574 | 0.9272 | 0.9777 | 0.9296 |
161
+ | 0.2572 | 87.0 | 5220 | 0.3426 | 0.9273 | 0.9781 | 0.9291 |
162
+ | 0.2646 | 88.0 | 5280 | 0.3472 | 0.9234 | 0.9789 | 0.9244 |
163
+ | 0.2831 | 89.0 | 5340 | 0.3433 | 0.9272 | 0.9779 | 0.9291 |
164
+ | 0.277 | 90.0 | 5400 | 0.3441 | 0.9263 | 0.9789 | 0.9280 |
165
+ | 0.2584 | 91.0 | 5460 | 0.3432 | 0.9236 | 0.9788 | 0.9249 |
166
+ | 0.2703 | 92.0 | 5520 | 0.3409 | 0.9248 | 0.9789 | 0.9259 |
167
+ | 0.2811 | 93.0 | 5580 | 0.3449 | 0.9215 | 0.9795 | 0.9228 |
168
+ | 0.2786 | 94.0 | 5640 | 0.3465 | 0.9260 | 0.9789 | 0.9280 |
169
+ | 0.267 | 95.0 | 5700 | 0.3472 | 0.9260 | 0.9791 | 0.9275 |
170
+ | 0.2695 | 96.0 | 5760 | 0.3500 | 0.9268 | 0.9786 | 0.9285 |
171
+ | 0.279 | 97.0 | 5820 | 0.3582 | 0.9249 | 0.9782 | 0.9270 |
172
+ | 0.2774 | 98.0 | 5880 | 0.3486 | 0.9251 | 0.9790 | 0.9270 |
173
+ | 0.2512 | 99.0 | 5940 | 0.3514 | 0.9287 | 0.9786 | 0.9306 |
174
+ | 0.2218 | 100.0 | 6000 | 0.3482 | 0.9269 | 0.9789 | 0.9285 |
175
+
176
+
177
+ ### Framework versions
178
+
179
+ - Transformers 4.29.2
180
+ - Pytorch 2.0.0+cu118
181
+ - Datasets 2.12.0
182
+ - Tokenizers 0.13.3