End of training

README.md

@@ -15,9 +15,9 @@ 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.3733
-- Validation Loss: 0.3217
-- Train Accuracy: 0.917
+- Train Loss: 0.3744
+- Validation Loss: 0.3587
+- Train Accuracy: 0.911
 - Epoch: 4
 
 ## Model description
@@ -44,11 +44,11 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.8079     | 1.6300          | 0.836          | 0     |
-| 1.2075     | 0.7969          | 0.894          | 1     |
-| 0.6799     | 0.5199          | 0.906          | 2     |
-| 0.4868     | 0.4028          | 0.899          | 3     |
-| 0.3733     | 0.3217          | 0.917          | 4     |
+| 2.7892     | 1.6801          | 0.793          | 0     |
+| 1.2251     | 0.8180          | 0.897          | 1     |
+| 0.6916     | 0.5436          | 0.891          | 2     |
+| 0.4950     | 0.3739          | 0.924          | 3     |
+| 0.3744     | 0.3587          | 0.911          | 4     |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eb55e52c4dd9cb54be719686034ae270936045ed3019056b9d008b8313d3b9d2
-size 343790264
+oid sha256:deec298b9c9f4c61df52476227dbf308d74baaec9a798c44b7a8341892b89f1d
+size 343773880