Model save

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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2577
-- Accuracy: 0.9425
+- Loss: 0.1836
+- Accuracy: 0.9575
 
 ## Model description
 
@@ -37,7 +37,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0003
+- learning_rate: 0.0002
 - train_batch_size: 4
 - eval_batch_size: 4
 - seed: 42
@@ -52,22 +52,22 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.1893        | 1.0   | 200  | 0.9100          | 0.69     |
-| 0.7705        | 2.0   | 400  | 0.6857          | 0.785    |
-| 0.6253        | 3.0   | 600  | 0.8546          | 0.7475   |
-| 0.5491        | 4.0   | 800  | 0.4918          | 0.8275   |
-| 0.4048        | 5.0   | 1000 | 0.4458          | 0.875    |
-| 0.3637        | 6.0   | 1200 | 0.3289          | 0.895    |
-| 0.263         | 7.0   | 1400 | 0.3888          | 0.86     |
-| 0.2325        | 8.0   | 1600 | 0.4741          | 0.8875   |
-| 0.196         | 9.0   | 1800 | 0.4416          | 0.8875   |
-| 0.2049        | 10.0  | 2000 | 0.2791          | 0.92     |
-| 0.1549        | 11.0  | 2200 | 0.2789          | 0.925    |
-| 0.126         | 12.0  | 2400 | 0.2609          | 0.93     |
-| 0.1378        | 13.0  | 2600 | 0.2549          | 0.935    |
-| 0.0884        | 14.0  | 2800 | 0.3017          | 0.93     |
-| 0.0907        | 15.0  | 3000 | 0.2586          | 0.93     |
-| 0.0707        | 16.0  | 3200 | 0.2577          | 0.9425   |
+| 1.1241        | 1.0   | 200  | 0.7452          | 0.765    |
+| 0.5854        | 2.0   | 400  | 0.4880          | 0.835    |
+| 0.4279        | 3.0   | 600  | 0.5049          | 0.8375   |
+| 0.4041        | 4.0   | 800  | 0.3321          | 0.8975   |
+| 0.2805        | 5.0   | 1000 | 0.4105          | 0.895    |
+| 0.279         | 6.0   | 1200 | 0.4269          | 0.8825   |
+| 0.1782        | 7.0   | 1400 | 0.3583          | 0.905    |
+| 0.1834        | 8.0   | 1600 | 0.3009          | 0.925    |
+| 0.1197        | 9.0   | 1800 | 0.3020          | 0.93     |
+| 0.1231        | 10.0  | 2000 | 0.3352          | 0.9225   |
+| 0.1273        | 11.0  | 2200 | 0.2908          | 0.91     |
+| 0.1019        | 12.0  | 2400 | 0.2528          | 0.94     |
+| 0.0951        | 13.0  | 2600 | 0.2989          | 0.9325   |
+| 0.0957        | 14.0  | 2800 | 0.3189          | 0.9325   |
+| 0.0618        | 15.0  | 3000 | 0.1973          | 0.9475   |
+| 0.0583        | 16.0  | 3200 | 0.1836          | 0.9575   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1f8434b843494c2059ac29ca083ec5750288fa07ecf5f2c361b404df390969de
+oid sha256:bfbdb5508a2bf8041a61d0f4216c8e415b6cd1a9e868cd7383e4ba0272ea3fc7
 size 343248584

runs/Mar20_07-15-37_d04eab29c3a2/events.out.tfevents.1710918938.d04eab29c3a2.379.6

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4d1067e1f8d50155f60438ecbdddbfd2bc751592de360dd4f2744ab922c53227
-size 11895
+oid sha256:8c6f361150b7e0e38572fca027ea8e9e92d735099e133567b2305f40a12bc2b7
+size 13851