Model save

README.md

@@ -2,7 +2,6 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
-- image-classification
 - generated_from_trainer
 datasets:
 - imagefolder
@@ -15,7 +14,7 @@ model-index:
       name: Image Classification
       type: image-classification
     dataset:
-      name: action_class
+      name: imagefolder
       type: imagefolder
       config: default
       split: train
@@ -23,7 +22,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.7885714285714286
+      value: 0.8095238095238095
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Action_model
 
-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 action_class 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 imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6974
-- Accuracy: 0.7886
+- Loss: 0.7479
+- Accuracy: 0.8095
 
 ## Model description
 
@@ -59,19 +58,16 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 2
+- num_epochs: 1
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.4296        | 0.32  | 100  | 0.8159          | 0.7448   |
-| 0.4083        | 0.64  | 200  | 0.6974          | 0.7886   |
-| 0.3384        | 0.96  | 300  | 0.8743          | 0.7371   |
-| 0.2812        | 1.27  | 400  | 0.7739          | 0.7676   |
-| 0.251         | 1.59  | 500  | 0.7166          | 0.7857   |
-| 0.2425        | 1.91  | 600  | 0.7692          | 0.7743   |
+| 1.1778        | 0.32  | 100  | 1.0651          | 0.78     |
+| 0.8527        | 0.64  | 200  | 0.7874          | 0.8133   |
+| 0.7114        | 0.96  | 300  | 0.7479          | 0.8095   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:09ece9292430947c4d04a44fb463dfd30531d4ca2b5b3af2515c644cdbc73d2c
+oid sha256:495d9650a1225c1a09df5da5257c925904dbe42b1d990d815beaa3303fb5a493
 size 343248584

runs/Apr16_13-47-24_a962527840ae/events.out.tfevents.1713275300.a962527840ae.34.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9dd184b56d428e06f987b02aa2cb16c4cb9045e0dfae8b0cfca968b197016da1
+oid sha256:aff72404cdfa2337f8499c3e9f3bea705172477c2a256987105d44755dddcae2
 size 4920