End of training

README.md

@@ -17,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 the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.2530
-- Accuracy: 0.4806
+- Loss: 1.0362
+- Accuracy: 0.6078
 
 ## Model description
 
@@ -50,13 +50,13 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.0298        | 1.0   | 17   | 1.8008          | 0.3428   |
-| 1.6758        | 2.0   | 34   | 1.5297          | 0.4276   |
-| 1.4213        | 3.0   | 51   | 1.3307          | 0.4523   |
-| 1.3225        | 4.0   | 68   | 1.2899          | 0.4806   |
-| 1.2432        | 5.0   | 85   | 1.2530          | 0.4806   |
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 1.4035        | 0.9767 | 21   | 1.2942          | 0.4735   |
+| 1.2173        | 2.0    | 43   | 1.1946          | 0.4382   |
+| 1.1276        | 2.9767 | 64   | 1.1574          | 0.5300   |
+| 1.0841        | 4.0    | 86   | 1.1034          | 0.5336   |
+| 1.0692        | 4.8837 | 105  | 1.0362          | 0.6078   |
 
 
 ### Framework versions