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: 1.9145
+- Loss: 1.9011
 - Accuracy: 0.9605
 
 ## Model description
@@ -64,9 +64,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 40   | 1.9285          | 0.9605   |
-| No log        | 2.0   | 80   | 1.6904          | 0.9605   |
-| No log        | 3.0   | 120  | 1.6333          | 0.9605   |
+| No log        | 1.0   | 40   | 1.8906          | 0.9605   |
+| No log        | 2.0   | 80   | 1.6868          | 0.9605   |
+| No log        | 3.0   | 120  | 1.6471          | 0.9605   |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bbd1b096e4dbdc60855e633e069aea9dc6bfbae177f5180e077f36c53823eb64
+oid sha256:b915555aa93b83f2dcda94bc4ce21e419778dc5487e733c0d9704ab3f539a78d
 size 343287149