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.8576449912126538
+      value: 0.843585237258348
 ---
 
 <!-- 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.4589
-- Accuracy: 0.8576
+- Loss: 0.6087
+- Accuracy: 0.8436
 
 ## Model description
 
@@ -59,44 +58,23 @@ 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: 10
+- num_epochs: 2
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.2754        | 0.37  | 100  | 1.1163          | 0.7329   |
-| 0.9345        | 0.75  | 200  | 0.8296          | 0.7996   |
-| 0.8816        | 1.12  | 300  | 0.7156          | 0.8102   |
-| 0.7425        | 1.49  | 400  | 0.6529          | 0.8067   |
-| 0.6883        | 1.87  | 500  | 0.6079          | 0.8243   |
-| 0.5454        | 2.24  | 600  | 0.5605          | 0.8348   |
-| 0.5383        | 2.61  | 700  | 0.5571          | 0.8295   |
-| 0.5442        | 2.99  | 800  | 0.5864          | 0.8190   |
-| 0.3986        | 3.36  | 900  | 0.5632          | 0.8313   |
-| 0.3438        | 3.73  | 1000 | 0.5606          | 0.8366   |
-| 0.4345        | 4.1   | 1100 | 0.5354          | 0.8366   |
-| 0.4523        | 4.48  | 1200 | 0.4988          | 0.8576   |
-| 0.3162        | 4.85  | 1300 | 0.5099          | 0.8541   |
-| 0.3793        | 5.22  | 1400 | 0.5190          | 0.8436   |
-| 0.3228        | 5.6   | 1500 | 0.4589          | 0.8576   |
-| 0.1795        | 5.97  | 1600 | 0.5096          | 0.8489   |
-| 0.2626        | 6.34  | 1700 | 0.5403          | 0.8489   |
-| 0.3041        | 6.72  | 1800 | 0.4908          | 0.8489   |
-| 0.1831        | 7.09  | 1900 | 0.5721          | 0.8383   |
-| 0.2275        | 7.46  | 2000 | 0.5349          | 0.8313   |
-| 0.1762        | 7.84  | 2100 | 0.5204          | 0.8541   |
-| 0.2112        | 8.21  | 2200 | 0.5189          | 0.8629   |
-| 0.1242        | 8.58  | 2300 | 0.5377          | 0.8471   |
-| 0.1207        | 8.96  | 2400 | 0.5325          | 0.8559   |
-| 0.1806        | 9.33  | 2500 | 0.5150          | 0.8647   |
-| 0.1793        | 9.7   | 2600 | 0.5153          | 0.8664   |
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 1.2783        | 0.3731 | 100  | 1.2065          | 0.7153   |
+| 0.9907        | 0.7463 | 200  | 0.8331          | 0.8102   |
+| 0.8428        | 1.1194 | 300  | 0.7278          | 0.8260   |
+| 0.7442        | 1.4925 | 400  | 0.6576          | 0.8172   |
+| 0.6749        | 1.8657 | 500  | 0.6087          | 0.8436   |
 
 
 ### Framework versions
 
-- Transformers 4.39.3
+- Transformers 4.41.1
 - Pytorch 2.1.2
-- Datasets 2.18.0
-- Tokenizers 0.15.2
+- Datasets 2.19.1
+- Tokenizers 0.19.1

config.json

@@ -44,5 +44,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.39.3"
+  "transformers_version": "4.41.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8b0a199ad74ff65ea529793bc6fdbef69fb6f7b70ddf2146d94e4c125b807dd2
+oid sha256:f500fee33cf67c06f93219817b2deb712e53e414d594bd9408c3b2249c73814d
 size 343248584

runs/Jun01_14-51-37_6c3c2d6efe19/events.out.tfevents.1717256949.6c3c2d6efe19.34.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3c8c7d4932a995756b5081ceec90e99008b51ea7166830884a36b9004ee45428
-size 4920
+oid sha256:0fb77d5f97f2e5ab7fa7e8770b92c00a16f8ca99d73dd63690a3e5b26e5212e6
+size 5048