End of training

README.md

@@ -14,9 +14,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.3796
-- Validation Loss: 0.3700
-- Train Accuracy: 0.901
+- Train Loss: 0.3923
+- Validation Loss: 0.3608
+- Train Accuracy: 0.904
 - Epoch: 4
 
 ## Model description
@@ -43,11 +43,11 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.8201     | 1.6921          | 0.804          | 0     |
-| 1.2229     | 0.8619          | 0.868          | 1     |
-| 0.7000     | 0.5735          | 0.882          | 2     |
-| 0.4806     | 0.4192          | 0.903          | 3     |
-| 0.3796     | 0.3700          | 0.901          | 4     |
+| 2.8080     | 1.6465          | 0.817          | 0     |
+| 1.2437     | 0.8525          | 0.878          | 1     |
+| 0.7138     | 0.5172          | 0.908          | 2     |
+| 0.5087     | 0.3884          | 0.91           | 3     |
+| 0.3923     | 0.3608          | 0.904          | 4     |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c83365eafe589520fff8310ca983b02303ba64afdfc3f66beb0ffb5cfb20dd09
+oid sha256:8c6f2b17e88c9069035b106702a8a688c556773d97c2956f0a2823cc71489df3
 size 343773880