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: vit_model
   results: []
@@ -14,7 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit_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 beans 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.0388
+- Accuracy: 0.9925
 
 ## Model description
 
@@ -41,6 +45,13 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 4
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 0.1321        | 3.8462 | 500  | 0.0388          | 0.9925   |
+
+
 ### Framework versions
 
 - Transformers 4.40.0

runs/Apr29_00-46-58_f6204f80b109/events.out.tfevents.1714351628.f6204f80b109.655.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0b54a082a41dad2441153c1157ce2bfe8691e6657ae004c10100c214db7c482d
-size 5281
+oid sha256:37aac9807297d9fe289488175238b8507f883db8773a86abb9b7ac80e18e966b
+size 5635