Model save

README.md

@@ -18,7 +18,7 @@ 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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0109
+- Loss: 0.0947
 - Accuracy: 1.0
 
 ## Model description
@@ -47,22 +47,17 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 10
+- num_epochs: 5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.366         | 1.0   | 4    | 0.9543          | 0.7857   |
-| 0.7687        | 2.0   | 8    | 0.4533          | 0.8571   |
-| 0.3512        | 3.0   | 12   | 0.2136          | 0.9464   |
-| 0.1623        | 4.0   | 16   | 0.0931          | 0.9821   |
-| 0.0987        | 5.0   | 20   | 0.0693          | 0.9643   |
-| 0.0688        | 6.0   | 24   | 0.0327          | 0.9821   |
-| 0.0846        | 7.0   | 28   | 0.0143          | 1.0      |
-| 0.0283        | 8.0   | 32   | 0.0165          | 1.0      |
-| 0.0413        | 9.0   | 36   | 0.0123          | 1.0      |
-| 0.031         | 10.0  | 40   | 0.0109          | 1.0      |
+| 1.291         | 1.0   | 4    | 0.8032          | 0.8393   |
+| 0.6667        | 2.0   | 8    | 0.3500          | 0.9643   |
+| 0.3311        | 3.0   | 12   | 0.1800          | 0.9464   |
+| 0.1914        | 4.0   | 16   | 0.1049          | 1.0      |
+| 0.146         | 5.0   | 20   | 0.0947          | 1.0      |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3c93944cf4de6b09a00cba758a52dc641ba1f10becfc37fe3c985e2b3433e463
+oid sha256:644432ef51b695439c039ab821e2eaf332b7e12b6bbf7bd99c431f1a5101e0c2
 size 110348984