End of training

README.md

@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.525
+      value: 0.59375
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,8 +33,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [ayooke97/emotion_classifier](https://huggingface.co/ayooke97/emotion_classifier) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.4129
-- Accuracy: 0.525
+- Loss: 1.1981
+- Accuracy: 0.5938
 
 ## Model description
 
@@ -67,15 +67,13 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 40   | 1.5552          | 0.4688   |
-| No log        | 2.0   | 80   | 1.4906          | 0.5      |
-| No log        | 3.0   | 120  | 1.4549          | 0.5062   |
-| No log        | 4.0   | 160  | 1.4412          | 0.5125   |
-| No log        | 5.0   | 200  | 1.4098          | 0.5125   |
-| No log        | 6.0   | 240  | 1.4129          | 0.525    |
-| No log        | 7.0   | 280  | 1.3911          | 0.525    |
-| No log        | 8.0   | 320  | 1.3882          | 0.5062   |
-| No log        | 9.0   | 360  | 1.4028          | 0.5188   |
+| No log        | 1.0   | 40   | 1.2351          | 0.575    |
+| No log        | 2.0   | 80   | 1.2232          | 0.575    |
+| No log        | 3.0   | 120  | 1.2373          | 0.55     |
+| No log        | 4.0   | 160  | 1.1981          | 0.5938   |
+| No log        | 5.0   | 200  | 1.1942          | 0.5687   |
+| No log        | 6.0   | 240  | 1.2105          | 0.5813   |
+| No log        | 7.0   | 280  | 1.1909          | 0.5813   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e133debfe0e6784cd22be27da09053ec1e123993895dd4df34ba839d6c0ae390
+oid sha256:12f204b0526c04568711360b5cfd17f4368371b44f82f4ebab8ab6ce7fedc191
 size 343242432

preprocessor_config.json

@@ -0,0 +1,23 @@
+{
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}