End of training

README.md

@@ -22,7 +22,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.8158995815899581
+      value: 0.8242677824267782
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,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 the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4121
-- Accuracy: 0.8159
+- Loss: 0.4274
+- Accuracy: 0.8243
 
 ## Model description
 
@@ -56,8 +56,8 @@ The following hyperparameters were used during training:
 - train_batch_size: 32
 - eval_batch_size: 32
 - seed: 69
-- gradient_accumulation_steps: 8
-- total_train_batch_size: 256
+- gradient_accumulation_steps: 16
+- total_train_batch_size: 512
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.05
@@ -68,20 +68,18 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.6839        | 0.89  | 15   | 0.6438          | 0.6757   |
-| 0.5555        | 1.78  | 30   | 0.5198          | 0.7364   |
-| 0.4995        | 2.67  | 45   | 0.5212          | 0.7469   |
-| 0.4177        | 3.56  | 60   | 0.4447          | 0.7866   |
-| 0.415         | 4.44  | 75   | 0.4438          | 0.7929   |
-| 0.3737        | 5.33  | 90   | 0.4302          | 0.7866   |
-| 0.3588        | 6.22  | 105  | 0.4452          | 0.7992   |
-| 0.3343        | 7.11  | 120  | 0.4666          | 0.7908   |
-| 0.3095        | 8.0   | 135  | 0.4727          | 0.7720   |
-| 0.2951        | 8.89  | 150  | 0.4162          | 0.8138   |
-| 0.2819        | 9.78  | 165  | 0.4299          | 0.8159   |
-| 0.257         | 10.67 | 180  | 0.4497          | 0.8033   |
-| 0.2625        | 11.56 | 195  | 0.4642          | 0.7971   |
-| 0.2287        | 12.44 | 210  | 0.4121          | 0.8159   |
+| 0.6782        | 1.78  | 15   | 0.5922          | 0.7008   |
+| 0.5096        | 3.56  | 30   | 0.5153          | 0.7552   |
+| 0.4434        | 5.33  | 45   | 0.4520          | 0.7762   |
+| 0.3844        | 7.11  | 60   | 0.4381          | 0.8013   |
+| 0.3642        | 8.89  | 75   | 0.4359          | 0.8054   |
+| 0.322         | 10.67 | 90   | 0.4086          | 0.8138   |
+| 0.2845        | 12.44 | 105  | 0.4111          | 0.8201   |
+| 0.2588        | 14.22 | 120  | 0.4100          | 0.8159   |
+| 0.2516        | 16.0  | 135  | 0.4122          | 0.8389   |
+| 0.2375        | 17.78 | 150  | 0.4085          | 0.8243   |
+| 0.2309        | 19.56 | 165  | 0.4149          | 0.8117   |
+| 0.2175        | 21.33 | 180  | 0.4274          | 0.8243   |
 
 
 ### Framework versions

model.safetensors

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