End of training

README.md

@@ -17,7 +17,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: 0.5563
+- Loss: 0.5429
 
 ## Model description
 
@@ -40,8 +40,6 @@ The following hyperparameters were used during training:
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 69
-- gradient_accumulation_steps: 8
-- total_train_batch_size: 128
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.1
@@ -51,22 +49,38 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 0.6302        | 0.88  | 20   | 0.5969          |
-| 0.5182        | 1.76  | 40   | 0.5321          |
-| 0.4872        | 2.64  | 60   | 0.5341          |
-| 0.4641        | 3.52  | 80   | 0.5578          |
-| 0.3737        | 4.4   | 100  | 0.5354          |
-| 0.2761        | 5.27  | 120  | 0.5345          |
-| 0.3207        | 6.15  | 140  | 0.5338          |
-| 0.391         | 7.03  | 160  | 0.5245          |
-| 0.282         | 7.91  | 180  | 0.5035          |
-| 0.2945        | 8.79  | 200  | 0.5162          |
-| 0.2355        | 9.67  | 220  | 0.5400          |
-| 0.2032        | 10.55 | 240  | 0.5418          |
-| 0.1749        | 11.43 | 260  | 0.5516          |
-| 0.2126        | 12.31 | 280  | 0.5357          |
-| 0.1701        | 13.19 | 300  | 0.5828          |
-| 0.2586        | 14.07 | 320  | 0.5563          |
+| 0.7599        | 0.11  | 20   | 0.7105          |
+| 0.6535        | 0.22  | 40   | 0.6561          |
+| 0.6191        | 0.33  | 60   | 0.6045          |
+| 0.7219        | 0.44  | 80   | 0.5976          |
+| 0.62          | 0.55  | 100  | 0.5781          |
+| 0.5249        | 0.66  | 120  | 0.5743          |
+| 0.5934        | 0.77  | 140  | 0.5684          |
+| 0.4143        | 0.88  | 160  | 0.5522          |
+| 0.4472        | 0.99  | 180  | 0.5645          |
+| 0.3507        | 1.1   | 200  | 0.5784          |
+| 0.4165        | 1.21  | 220  | 0.5796          |
+| 0.2822        | 1.32  | 240  | 0.5867          |
+| 0.4765        | 1.43  | 260  | 0.5446          |
+| 0.4557        | 1.54  | 280  | 0.5739          |
+| 0.4087        | 1.65  | 300  | 0.5292          |
+| 0.4181        | 1.76  | 320  | 0.5182          |
+| 0.6616        | 1.87  | 340  | 0.5328          |
+| 0.48          | 1.98  | 360  | 0.5282          |
+| 0.2159        | 2.09  | 380  | 0.5457          |
+| 0.5038        | 2.2   | 400  | 0.5832          |
+| 0.6434        | 2.31  | 420  | 0.5667          |
+| 0.4566        | 2.42  | 440  | 0.5128          |
+| 0.2972        | 2.53  | 460  | 0.5475          |
+| 0.5094        | 2.64  | 480  | 0.5536          |
+| 0.4807        | 2.75  | 500  | 0.5512          |
+| 0.3666        | 2.86  | 520  | 0.5128          |
+| 0.38          | 2.97  | 540  | 0.5499          |
+| 0.3496        | 3.08  | 560  | 0.5129          |
+| 0.3503        | 3.19  | 580  | 0.5480          |
+| 0.4435        | 3.3   | 600  | 0.5918          |
+| 0.2163        | 3.41  | 620  | 0.5558          |
+| 0.4316        | 3.52  | 640  | 0.5429          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c8c233697547edb87083fafd2280aba1af15a0c10da5a09c094becbc5e6114ee
+oid sha256:ff67574d4ea45745b08d418e8e3e4a5a7cea01447211226063ecdf588d5f0933
 size 343223968