🍻 cheers

README.md

@@ -2,6 +2,7 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
+- image-classification
 - generated_from_trainer
 datasets:
 - imagefolder
@@ -14,7 +15,7 @@ model-index:
       name: Image Classification
       type: image-classification
     dataset:
-      name: imagefolder
+      name: agent_action_class
       type: imagefolder
       config: default
       split: train
@@ -22,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.7342857142857143
+      value: 0.7628571428571429
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,11 +31,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Action_Classification
 
-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.
+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 agent_action_class dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.3112
-- Accuracy: 0.7343
-- Confusion Matrix: [[66, 3, 9, 5, 0, 2, 3, 3, 7, 2], [52, 98, 0, 0, 2, 1, 6, 4, 30, 0], [3, 1, 52, 1, 1, 3, 2, 0, 0, 2], [4, 0, 5, 33, 5, 0, 0, 0, 0, 5], [0, 2, 1, 3, 83, 1, 2, 1, 0, 0], [2, 2, 13, 4, 0, 44, 2, 2, 1, 2], [0, 0, 1, 0, 1, 0, 56, 0, 0, 1], [2, 0, 2, 0, 0, 0, 1, 78, 0, 0], [9, 28, 0, 0, 1, 1, 15, 1, 124, 1], [3, 0, 1, 10, 0, 1, 0, 1, 0, 137]]
+- Loss: 0.8482
+- Accuracy: 0.7629
+- Confusion Matrix: [[45, 5, 20, 4, 2, 6, 4, 8, 3, 3], [5, 154, 4, 2, 1, 2, 6, 1, 17, 1], [0, 0, 51, 1, 2, 8, 1, 0, 0, 2], [1, 0, 8, 26, 8, 5, 0, 0, 1, 3], [0, 1, 0, 0, 89, 3, 0, 0, 0, 0], [0, 1, 11, 3, 1, 55, 0, 1, 0, 0], [0, 1, 1, 0, 3, 3, 51, 0, 0, 0], [0, 0, 10, 1, 0, 4, 0, 68, 0, 0], [0, 26, 5, 0, 1, 3, 16, 1, 127, 1], [3, 0, 2, 9, 2, 1, 0, 1, 0, 135]]
 
 ## Model description
 

all_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 15.0,
+    "total_flos": 1.1677717704563528e+19,
+    "train_loss": 0.19079542797209098,
+    "train_runtime": 4656.9657,
+    "train_samples_per_second": 32.323,
+    "train_steps_per_second": 1.011
+}

runs/Apr26_19-40-54_2fba0b6b6ec2/events.out.tfevents.1714165396.2fba0b6b6ec2.34.2

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

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 15.0,
+    "total_flos": 1.1677717704563528e+19,
+    "train_loss": 0.19079542797209098,
+    "train_runtime": 4656.9657,
+    "train_samples_per_second": 32.323,
+    "train_steps_per_second": 1.011
+}