Model save

README.md

@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.9651567944250871
+      value: 0.9372822299651568
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,8 +33,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1167
-- Accuracy: 0.9652
+- Loss: 0.1636
+- Accuracy: 0.9373
 
 ## Model description
 
@@ -54,12 +54,12 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 16
-- eval_batch_size: 16
+- train_batch_size: 32
+- eval_batch_size: 32
 - seed: 42
 - gradient_accumulation_steps: 4
-- total_train_batch_size: 64
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- total_train_batch_size: 128
+- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
 - num_epochs: 5
@@ -68,16 +68,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| 0.0628        | 0.9877 | 40   | 0.1494          | 0.9512   |
-| 0.0499        | 2.0    | 81   | 0.1830          | 0.9373   |
-| 0.0697        | 2.9877 | 121  | 0.1094          | 0.9547   |
-| 0.0646        | 4.0    | 162  | 0.1318          | 0.9582   |
-| 0.0481        | 4.9383 | 200  | 0.1167          | 0.9652   |
+| 0.8509        | 0.9877 | 20   | 0.5305          | 0.8467   |
+| 0.4478        | 1.9753 | 40   | 0.3092          | 0.9094   |
+| 0.3313        | 2.9630 | 60   | 0.2422          | 0.9233   |
+| 0.2777        | 4.0    | 81   | 0.1716          | 0.9373   |
+| 0.2465        | 4.9383 | 100  | 0.1636          | 0.9373   |
 
 
 ### Framework versions
 
-- Transformers 4.44.2
+- Transformers 4.46.0.dev0
 - Pytorch 2.4.1+cu121
 - Datasets 3.0.1
-- Tokenizers 0.19.1
+- Tokenizers 0.20.1

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bf90d940797d14391f324bcdc7d013ceb5f346ab0f97a8c5716686934f8d256e
+oid sha256:035707ba7def84500aaa8a6501235bc2da59c7aa8f09dcb90e5701c17922b7e4
 size 110348984