End of training

README.md

@@ -0,0 +1,81 @@
+---
+library_name: transformers
+license: apache-2.0
+base_model: google/vit-base-patch16-224-in21k
+tags:
+- generated_from_trainer
+datasets:
+- imagefolder
+metrics:
+- accuracy
+model-index:
+- name: emotion_recognition
+  results:
+  - task:
+      name: Image Classification
+      type: image-classification
+    dataset:
+      name: imagefolder
+      type: imagefolder
+      config: default
+      split: train
+      args: default
+    metrics:
+    - name: Accuracy
+      type: accuracy
+      value: 0.26875
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# emotion_recognition
+
+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.
+It achieves the following results on the evaluation set:
+- Loss: 1.9225
+- Accuracy: 0.2687
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5e-05
+- train_batch_size: 16
+- eval_batch_size: 16
+- seed: 42
+- gradient_accumulation_steps: 4
+- total_train_batch_size: 64
+- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 3
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 2.053         | 1.0   | 10   | 2.0367          | 0.1812   |
+| 1.9519        | 2.0   | 20   | 1.9521          | 0.2812   |
+| 1.8648        | 3.0   | 30   | 1.9228          | 0.2938   |
+
+
+### Framework versions
+
+- Transformers 4.48.3
+- Pytorch 2.5.1+cu124
+- Datasets 3.3.2
+- Tokenizers 0.21.0

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:924e9696fe6aa60f6d5744934b34084a398499e927bd581583e27c1a180670c4
+oid sha256:549b44c7c8f6691d2078d8074104e6d58e7c4e38db69a02179ef39261fe547c4
 size 343242432

runs/Feb24_02-55-24_6c9b8eea2d51/events.out.tfevents.1740370517.6c9b8eea2d51.12403.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47e599879d4642f441c7f4bffec1740e4e60b3ec12b6160056e0e9e0472d88fa
+size 405

runs/Feb24_04-37-29_6c9b8eea2d51/events.out.tfevents.1740371914.6c9b8eea2d51.12403.2

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

runs/Feb24_04-41-35_6c9b8eea2d51/events.out.tfevents.1740372120.6c9b8eea2d51.38221.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:124fc0691a019d409a7e00af0db1b18cea61048b7cd8153488b7a2c7ad11439c
+size 12814

runs/Feb24_04-41-35_6c9b8eea2d51/events.out.tfevents.1740377797.6c9b8eea2d51.38221.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:91ccb390c4b0c93e9128d90cd43f956da7153ae0f2eaa105e594f980cbc110ab
+size 405

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:882dc668229f30dfab6bc8a09e711b6491cd4442b8083a5083fd01155a0b0439
+oid sha256:1e1c4566987dd9158f91ad49190afee8ec0a4666c31dd1743b074627df058da5
 size 5368