End of training

README.md

@@ -1,18 +1,18 @@
 ---
+library_name: transformers
 base_model: nateraw/vit-age-classifier
+tags:
+- generated_from_trainer
 datasets:
 - imagefolder
-library_name: transformers
 metrics:
 - accuracy
-tags:
-- generated_from_trainer
 model-index:
 - name: results
   results:
   - task:
-      type: image-classification
       name: Image Classification
+      type: image-classification
     dataset:
       name: imagefolder
       type: imagefolder
@@ -20,9 +20,9 @@ model-index:
       split: train
       args: default
     metrics:
-    - type: accuracy
-      value: 0.44375
-      name: Accuracy
+    - name: Accuracy
+      type: accuracy
+      value: 0.5
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [nateraw/vit-age-classifier](https://huggingface.co/nateraw/vit-age-classifier) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.6666
-- Accuracy: 0.4437
+- Loss: 1.7301
+- Accuracy: 0.5
 
 ## Model description
 
@@ -64,9 +64,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.968         | 1.0   | 40   | 1.9059          | 0.3063   |
-| 1.5965        | 2.0   | 80   | 1.7152          | 0.3937   |
-| 1.4239        | 3.0   | 120  | 1.6666          | 0.4437   |
+| 2.0108        | 1.0   | 40   | 1.9813          | 0.325    |
+| 1.7144        | 2.0   | 80   | 1.8097          | 0.45     |
+| 1.5272        | 3.0   | 120  | 1.7301          | 0.5      |
 
 
 ### Framework versions

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "aneirady/results",
+  "_name_or_path": "nateraw/vit-age-classifier",
   "architectures": [
     "ViTForImageClassification"
   ],

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:55308629ced5b2367cb0dd43c7a0a83a5153e45a51e0070bf21cd9e4bdb636cc
+oid sha256:dfc54cbcf525067664bba15a50d4d73c033221e24585dbd1892280b01fa7ec92
 size 343242432