End of training

README.md

@@ -32,7 +32,7 @@ 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 the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4646
+- Loss: 0.4822
 - Accuracy: 0.8210
 
 ## Model description
@@ -65,14 +65,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.4933        | 1.16  | 150  | 0.5369          | 0.7183   |
-| 0.445         | 2.33  | 300  | 0.4573          | 0.7860   |
-| 0.3926        | 3.49  | 450  | 0.4551          | 0.7751   |
-| 0.2821        | 4.65  | 600  | 0.5172          | 0.7795   |
-| 0.2778        | 5.81  | 750  | 0.4400          | 0.7991   |
-| 0.2352        | 6.98  | 900  | 0.4533          | 0.8057   |
-| 0.2244        | 8.14  | 1050 | 0.4721          | 0.8079   |
-| 0.1838        | 9.3   | 1200 | 0.4646          | 0.8210   |
+| 0.4892        | 1.16  | 150  | 0.5660          | 0.7271   |
+| 0.4346        | 2.33  | 300  | 0.4475          | 0.7860   |
+| 0.4072        | 3.49  | 450  | 0.4499          | 0.7882   |
+| 0.312         | 4.65  | 600  | 0.5289          | 0.7598   |
+| 0.2913        | 5.81  | 750  | 0.4868          | 0.8100   |
+| 0.2389        | 6.98  | 900  | 0.4596          | 0.8079   |
+| 0.2176        | 8.14  | 1050 | 0.4642          | 0.8210   |
+| 0.183         | 9.3   | 1200 | 0.4822          | 0.8210   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aac5e4892d56994decd4670fc47a58fa0cd2a4760e15a83928bd6002d027f1a0
+oid sha256:89cb6e6cb08f72d0aeab3b00543e5ed8649288a2f46c98625f4f5f61ded94e9e
 size 343223968