Hugging Face's logo

mhdiqbalpradipta
/
emotion_classification

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results (legacy)
Model card Files
xet
Metrics Community
mhdiqbalpradipta commited on
Commit
aeff5c6
·
verified ·
1 Parent(s): 1f34114

End of training

Browse files
Files changed (2) hide show
  1. README.md +62 -62
  2. model.safetensors +1 -1
README.md CHANGED
@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 1.1749
36
  - Accuracy: 0.5813
37
 
38
  ## Model description
@@ -57,7 +57,7 @@ The following hyperparameters were used during training:
57
  - eval_batch_size: 16
58
  - seed: 42
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
- - lr_scheduler_type: linear
61
  - num_epochs: 60
62
  - mixed_precision_training: Native AMP
63
 
@@ -65,66 +65,66 @@ The following hyperparameters were used during training:
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
- | 2.073 | 1.0 | 40 | 2.0553 | 0.2188 |
69
- | 2.0238 | 2.0 | 80 | 2.0162 | 0.275 |
70
- | 1.9598 | 3.0 | 120 | 1.9458 | 0.4 |
71
- | 1.8585 | 4.0 | 160 | 1.8555 | 0.3937 |
72
- | 1.7579 | 5.0 | 200 | 1.7204 | 0.475 |
73
- | 1.6636 | 6.0 | 240 | 1.6270 | 0.4688 |
74
- | 1.5809 | 7.0 | 280 | 1.5691 | 0.525 |
75
- | 1.4996 | 8.0 | 320 | 1.5250 | 0.5062 |
76
- | 1.4555 | 9.0 | 360 | 1.4532 | 0.525 |
77
- | 1.4088 | 10.0 | 400 | 1.4374 | 0.5188 |
78
- | 1.3475 | 11.0 | 440 | 1.4162 | 0.5375 |
79
- | 1.3107 | 12.0 | 480 | 1.3727 | 0.525 |
80
- | 1.2669 | 13.0 | 520 | 1.3535 | 0.5375 |
81
- | 1.2375 | 14.0 | 560 | 1.3533 | 0.525 |
82
- | 1.1865 | 15.0 | 600 | 1.3284 | 0.5375 |
83
- | 1.156 | 16.0 | 640 | 1.3288 | 0.5312 |
84
- | 1.1148 | 17.0 | 680 | 1.2972 | 0.55 |
85
- | 1.0744 | 18.0 | 720 | 1.2742 | 0.5563 |
86
- | 1.0481 | 19.0 | 760 | 1.2473 | 0.575 |
87
- | 1.0008 | 20.0 | 800 | 1.2330 | 0.5875 |
88
- | 0.9788 | 21.0 | 840 | 1.2163 | 0.575 |
89
- | 0.9766 | 22.0 | 880 | 1.2781 | 0.5687 |
90
- | 0.933 | 23.0 | 920 | 1.2021 | 0.575 |
91
- | 0.8953 | 24.0 | 960 | 1.2426 | 0.5625 |
92
- | 0.8701 | 25.0 | 1000 | 1.1871 | 0.5813 |
93
- | 0.8647 | 26.0 | 1040 | 1.2654 | 0.5125 |
94
- | 0.8432 | 27.0 | 1080 | 1.2051 | 0.5437 |
95
- | 0.8128 | 28.0 | 1120 | 1.2658 | 0.5437 |
96
- | 0.8382 | 29.0 | 1160 | 1.2093 | 0.55 |
97
- | 0.7872 | 30.0 | 1200 | 1.2364 | 0.55 |
98
- | 0.7662 | 31.0 | 1240 | 1.1820 | 0.5875 |
99
- | 0.7354 | 32.0 | 1280 | 1.2133 | 0.5813 |
100
- | 0.7251 | 33.0 | 1320 | 1.1519 | 0.5938 |
101
- | 0.7003 | 34.0 | 1360 | 1.2387 | 0.5813 |
102
- | 0.6787 | 35.0 | 1400 | 1.2615 | 0.525 |
103
- | 0.6776 | 36.0 | 1440 | 1.2160 | 0.5813 |
104
- | 0.6486 | 37.0 | 1480 | 1.2137 | 0.5687 |
105
- | 0.6107 | 38.0 | 1520 | 1.2362 | 0.55 |
106
- | 0.6231 | 39.0 | 1560 | 1.1770 | 0.5625 |
107
- | 0.5947 | 40.0 | 1600 | 1.2345 | 0.5437 |
108
- | 0.6302 | 41.0 | 1640 | 1.1654 | 0.6125 |
109
- | 0.5881 | 42.0 | 1680 | 1.2213 | 0.5625 |
110
- | 0.6075 | 43.0 | 1720 | 1.2112 | 0.55 |
111
- | 0.581 | 44.0 | 1760 | 1.1680 | 0.6 |
112
- | 0.5646 | 45.0 | 1800 | 1.1939 | 0.6 |
113
- | 0.5306 | 46.0 | 1840 | 1.1687 | 0.6188 |
114
- | 0.5545 | 47.0 | 1880 | 1.1530 | 0.5687 |
115
- | 0.5585 | 48.0 | 1920 | 1.1791 | 0.5813 |
116
- | 0.5484 | 49.0 | 1960 | 1.2595 | 0.55 |
117
- | 0.5492 | 50.0 | 2000 | 1.2213 | 0.5437 |
118
- | 0.5276 | 51.0 | 2040 | 1.1498 | 0.5938 |
119
- | 0.5298 | 52.0 | 2080 | 1.1860 | 0.5875 |
120
- | 0.5006 | 53.0 | 2120 | 1.2137 | 0.5563 |
121
- | 0.522 | 54.0 | 2160 | 1.2012 | 0.5687 |
122
- | 0.5287 | 55.0 | 2200 | 1.1927 | 0.5312 |
123
- | 0.516 | 56.0 | 2240 | 1.1973 | 0.5375 |
124
- | 0.5 | 57.0 | 2280 | 1.1854 | 0.5687 |
125
- | 0.4906 | 58.0 | 2320 | 1.2936 | 0.5 |
126
- | 0.5329 | 59.0 | 2360 | 1.2269 | 0.5437 |
127
- | 0.5122 | 60.0 | 2400 | 1.1749 | 0.5813 |
128
 
129
 
130
  ### Framework versions
 
32
 
33
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 1.1601
36
  - Accuracy: 0.5813
37
 
38
  ## Model description
 
57
  - eval_batch_size: 16
58
  - seed: 42
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
+ - lr_scheduler_type: cosine_with_restarts
61
  - num_epochs: 60
62
  - mixed_precision_training: Native AMP
63
 
 
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 2.0632 | 1.0 | 40 | 2.0444 | 0.2437 |
69
+ | 2.0072 | 2.0 | 80 | 1.9824 | 0.375 |
70
+ | 1.9215 | 3.0 | 120 | 1.8844 | 0.4062 |
71
+ | 1.7969 | 4.0 | 160 | 1.7549 | 0.4688 |
72
+ | 1.6868 | 5.0 | 200 | 1.6343 | 0.4688 |
73
+ | 1.5946 | 6.0 | 240 | 1.5604 | 0.4938 |
74
+ | 1.5223 | 7.0 | 280 | 1.5055 | 0.525 |
75
+ | 1.4588 | 8.0 | 320 | 1.4665 | 0.5188 |
76
+ | 1.4157 | 9.0 | 360 | 1.3954 | 0.575 |
77
+ | 1.3713 | 10.0 | 400 | 1.3775 | 0.5312 |
78
+ | 1.3101 | 11.0 | 440 | 1.3572 | 0.5563 |
79
+ | 1.2881 | 12.0 | 480 | 1.3337 | 0.5437 |
80
+ | 1.2328 | 13.0 | 520 | 1.3253 | 0.525 |
81
+ | 1.2115 | 14.0 | 560 | 1.3186 | 0.5312 |
82
+ | 1.1531 | 15.0 | 600 | 1.3023 | 0.5375 |
83
+ | 1.1198 | 16.0 | 640 | 1.3149 | 0.5125 |
84
+ | 1.0833 | 17.0 | 680 | 1.2865 | 0.5312 |
85
+ | 1.0393 | 18.0 | 720 | 1.2498 | 0.525 |
86
+ | 1.0058 | 19.0 | 760 | 1.2166 | 0.5875 |
87
+ | 0.9721 | 20.0 | 800 | 1.1884 | 0.6 |
88
+ | 0.9387 | 21.0 | 840 | 1.1802 | 0.6 |
89
+ | 0.9505 | 22.0 | 880 | 1.2789 | 0.5125 |
90
+ | 0.8972 | 23.0 | 920 | 1.1719 | 0.5875 |
91
+ | 0.8665 | 24.0 | 960 | 1.2168 | 0.5625 |
92
+ | 0.8503 | 25.0 | 1000 | 1.1918 | 0.5437 |
93
+ | 0.8357 | 26.0 | 1040 | 1.2575 | 0.5062 |
94
+ | 0.8275 | 27.0 | 1080 | 1.1863 | 0.5938 |
95
+ | 0.7957 | 28.0 | 1120 | 1.2722 | 0.5375 |
96
+ | 0.8062 | 29.0 | 1160 | 1.1875 | 0.5687 |
97
+ | 0.7625 | 30.0 | 1200 | 1.2282 | 0.5625 |
98
+ | 0.7529 | 31.0 | 1240 | 1.1860 | 0.5687 |
99
+ | 0.721 | 32.0 | 1280 | 1.2366 | 0.55 |
100
+ | 0.7047 | 33.0 | 1320 | 1.1710 | 0.575 |
101
+ | 0.6803 | 34.0 | 1360 | 1.2395 | 0.5437 |
102
+ | 0.6858 | 35.0 | 1400 | 1.2439 | 0.5563 |
103
+ | 0.6613 | 36.0 | 1440 | 1.1809 | 0.5563 |
104
+ | 0.6453 | 37.0 | 1480 | 1.1892 | 0.5375 |
105
+ | 0.6052 | 38.0 | 1520 | 1.2089 | 0.5437 |
106
+ | 0.6193 | 39.0 | 1560 | 1.1421 | 0.5938 |
107
+ | 0.6026 | 40.0 | 1600 | 1.2163 | 0.5625 |
108
+ | 0.6345 | 41.0 | 1640 | 1.1429 | 0.5938 |
109
+ | 0.5842 | 42.0 | 1680 | 1.1989 | 0.5625 |
110
+ | 0.6326 | 43.0 | 1720 | 1.1958 | 0.6125 |
111
+ | 0.5933 | 44.0 | 1760 | 1.1793 | 0.5375 |
112
+ | 0.5902 | 45.0 | 1800 | 1.1795 | 0.5938 |
113
+ | 0.5497 | 46.0 | 1840 | 1.1797 | 0.5563 |
114
+ | 0.578 | 47.0 | 1880 | 1.1516 | 0.5813 |
115
+ | 0.5793 | 48.0 | 1920 | 1.1770 | 0.6 |
116
+ | 0.5857 | 49.0 | 1960 | 1.1991 | 0.5625 |
117
+ | 0.5742 | 50.0 | 2000 | 1.2873 | 0.5062 |
118
+ | 0.5696 | 51.0 | 2040 | 1.1254 | 0.6062 |
119
+ | 0.5745 | 52.0 | 2080 | 1.1817 | 0.6 |
120
+ | 0.5523 | 53.0 | 2120 | 1.2076 | 0.5875 |
121
+ | 0.5554 | 54.0 | 2160 | 1.1946 | 0.5938 |
122
+ | 0.5726 | 55.0 | 2200 | 1.1964 | 0.5813 |
123
+ | 0.5702 | 56.0 | 2240 | 1.1921 | 0.575 |
124
+ | 0.5396 | 57.0 | 2280 | 1.1801 | 0.6375 |
125
+ | 0.5444 | 58.0 | 2320 | 1.2866 | 0.5437 |
126
+ | 0.5713 | 59.0 | 2360 | 1.1860 | 0.5938 |
127
+ | 0.5626 | 60.0 | 2400 | 1.1601 | 0.5813 |
128
 
129
 
130
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a4c83416c3de18ead25f6b22a9c8462bde5659d452c0b0f0cf59e1f7ebee762d
3
  size 343242432
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7ec66ba338c5b393c24c0fb801012ab0ff5f2b215cd01abafea623385fdbb3f
3
  size 343242432