Model save

README.md

@@ -2,8 +2,9 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
-- image-classification
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: finetuned-indian-food
   results: []
@@ -14,7 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # finetuned-indian-food
 
-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 indian_food_images 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 an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2867
+- Accuracy: 0.9267
 
 ## Model description
 
@@ -42,6 +46,25 @@ The following hyperparameters were used during training:
 - num_epochs: 4
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 1.0192        | 0.3003 | 100  | 0.9248          | 0.8480   |
+| 0.635         | 0.6006 | 200  | 0.5917          | 0.8863   |
+| 0.6523        | 0.9009 | 300  | 0.5134          | 0.8799   |
+| 0.4247        | 1.2012 | 400  | 0.3983          | 0.9044   |
+| 0.4393        | 1.5015 | 500  | 0.4119          | 0.8980   |
+| 0.4631        | 1.8018 | 600  | 0.3752          | 0.9107   |
+| 0.2992        | 2.1021 | 700  | 0.3469          | 0.9129   |
+| 0.3           | 2.4024 | 800  | 0.3157          | 0.9203   |
+| 0.2372        | 2.7027 | 900  | 0.3210          | 0.9192   |
+| 0.2447        | 3.0030 | 1000 | 0.3140          | 0.9224   |
+| 0.2209        | 3.3033 | 1100 | 0.3034          | 0.9160   |
+| 0.2641        | 3.6036 | 1200 | 0.2896          | 0.9277   |
+| 0.0954        | 3.9039 | 1300 | 0.2867          | 0.9267   |
+
+
 ### Framework versions
 
 - Transformers 4.42.4

config.json

@@ -61,6 +61,7 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
+  "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.42.4"

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d58c073314f5ac2f7ca77bfb5ec45f829058974ecb27cd4cf55ff51c5658b04f
+oid sha256:e005b47bf0a3e628ba056c9d80df83a4157d3c57539c95c64b7f64ae9cdcdc97
 size 343279344

runs/Aug22_09-00-22_e6ee7653b19d/events.out.tfevents.1724317241.e6ee7653b19d.364.0

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

runs/Aug22_09-03-46_e6ee7653b19d/events.out.tfevents.1724317443.e6ee7653b19d.364.1

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:13f9ebd3c294e9aec77c38451c69c262cae96a32a8df36dab5148d0a6c05fdde
+oid sha256:98750c7900609da7837278d54129418c77f926c8b4dfb16a55dbbdd349afe395
 size 5112