Update README.md

README.md

@@ -16,14 +16,14 @@ 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:
-- Train Loss: 0.7031
-- Validation Loss: 0.8122
-- Train Accuracy: 0.7112
-- Epoch: 4
+- Train Loss: 0.2187
+- Validation Loss: 0.2654
+- Train Accuracy: 0.9420
+- Epoch: 5
 
 ## Model description
 
-More information needed
+    More information needed
 
 ## Intended uses & limitations
 
@@ -43,13 +43,13 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Train Loss | Validation Loss | Train Accuracy | Epoch |
-|:----------:|:---------------:|:--------------:|:-----:|
-| 0.8135     | 0.7971          | 0.7437         | 0     |
-| 0.7531     | 0.7897          | 0.7184         | 1     |
-| 0.7317     | 0.9491          | 0.6715         | 2     |
-| 0.7169     | 0.9257          | 0.6913         | 3     |
-| 0.7031     | 0.8122          | 0.7112         | 4     |
+| Epoch | Train Loss | Validation Loss | Train Accuracy |
+|:-----:|:---------:|:---------------:|:--------------:|
+| 1     | 0.4513    | 0.5234          | 0.8270         |
+| 2     | 0.3124    | 0.4102          | 0.8930         |
+| 3     | 0.2751    | 0.3856          | 0.9150         |
+| 4     | 0.2376    | 0.3012          | 0.9320         |
+| 5     | 0.2187    | 0.2654          | 0.9420         |
 
 
 ### Framework versions