End of training

README.md

@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [pradanaadn/vit-emotional-classifier](https://huggingface.co/pradanaadn/vit-emotional-classifier) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.3140
+- Loss: 1.1830
 - Accuracy: 0.6312
 
 ## Model description
@@ -53,8 +53,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 32
-- eval_batch_size: 8
+- train_batch_size: 16
+- eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine_with_restarts
@@ -64,11 +64,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.1403        | 1.0   | 20   | 1.4137          | 0.5938   |
-| 1.0421        | 2.0   | 40   | 1.3920          | 0.6188   |
-| 0.9611        | 3.0   | 60   | 1.3161          | 0.6      |
-| 0.8988        | 4.0   | 80   | 1.3140          | 0.6312   |
-| 0.8381        | 5.0   | 100  | 1.2554          | 0.6      |
+| 0.9412        | 0.5   | 20   | 1.3606          | 0.575    |
+| 0.8621        | 1.0   | 40   | 1.3134          | 0.6125   |
+| 0.8025        | 1.5   | 60   | 1.2917          | 0.6062   |
+| 0.7077        | 2.0   | 80   | 1.2553          | 0.6062   |
+| 0.7259        | 2.5   | 100  | 1.2128          | 0.625    |
+| 0.5685        | 3.0   | 120  | 1.2036          | 0.625    |
+| 0.5604        | 3.5   | 140  | 1.2057          | 0.6062   |
+| 0.4817        | 4.0   | 160  | 1.1830          | 0.6312   |
+| 0.4421        | 4.5   | 180  | 1.2004          | 0.5875   |
+| 0.4692        | 5.0   | 200  | 1.1568          | 0.6188   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9d820aa0397e5c3acf9b773f1544c9c7801b198429dc256d13022d211b6a92ba
+oid sha256:b45b541e3f75a71228d11ded546dc5272ad932075c88d57c9c37d2500f9dfa7f
 size 343242432

runs/Jun05_09-16-31_f8aa38a62fb7/events.out.tfevents.1717579321.f8aa38a62fb7.237.8

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3505ee6b1026d2609332442eefa1038c37d3c36ef2b69538c55e4c4529d7e716
+size 411