Model save

README.md

@@ -2,8 +2,6 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
-- image-classification
-- vision
 - generated_from_trainer
 metrics:
 - f1
@@ -19,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 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.0410
-- F1: 0.9705
+- Loss: 0.0460
+- F1: 0.9429
 
 ## Model description
 
@@ -49,12 +47,13 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | F1     |
-|:-------------:|:-----:|:----:|:---------------:|:------:|
-| 0.0663        | 1.0   | 1981 | 0.0594          | 0.9705 |
-| 0.0747        | 2.0   | 3962 | 0.0410          | 0.9705 |
-| 0.0676        | 3.0   | 5943 | 0.0463          | 0.9705 |
-| 0.0706        | 4.0   | 7924 | 0.0445          | 0.9705 |
+| Training Loss | Epoch | Step | F1     | Validation Loss |
+|:-------------:|:-----:|:----:|:------:|:---------------:|
+| 0.0663        | 1.0   | 1981 | 0.9705 | 0.0594          |
+| 0.0747        | 2.0   | 3962 | 0.9705 | 0.0410          |
+| 0.0676        | 3.0   | 5943 | 0.9705 | 0.0463          |
+| 0.0706        | 4.0   | 7924 | 0.9705 | 0.0445          |
+| 0.0595        | 5.0   | 9905 | 0.0460 | 0.9429          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0187ef65b16ea02034186535c521881523d76e613070c81c7da241a2b2330692
+oid sha256:572303f761b7a7387078fed0e6d4bb406e84528aa38cd5998d45fe55f1830673
 size 343220892