mmanikanta
/

SWIN-AI-Image-Detector

Image Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

mmanikanta commited on Apr 20

Commit

5b4602c

•

1 Parent(s): d72f8f9

update model card README.md

Files changed (1) hide show

README.md +14 -14

README.md CHANGED Viewed

@@ -14,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
 # SWIN-AI-Image-Detector
-This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on CIFAKE dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0342
-- Accuracy: 0.9874
 ## Model description
@@ -51,21 +51,21 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.2432        | 1.0   | 703  | 0.1859          | 0.9267   |
-| 0.1665        | 2.0   | 1406 | 0.1229          | 0.9514   |
-| 0.1427        | 3.0   | 2109 | 0.0604          | 0.9761   |
-| 0.143         | 4.0   | 2813 | 0.0807          | 0.97     |
-| 0.1008        | 5.0   | 3516 | 0.0444          | 0.9824   |
-| 0.1189        | 6.0   | 4219 | 0.0635          | 0.977    |
-| 0.0936        | 7.0   | 4922 | 0.0535          | 0.9789   |
-| 0.1201        | 8.0   | 5626 | 0.0376          | 0.9854   |
-| 0.076         | 9.0   | 6329 | 0.0342          | 0.9874   |
-| 0.0887        | 10.0  | 7030 | 0.0413          | 0.9844   |
 ### Framework versions
 - Transformers 4.30.0
 - Pytorch 2.2.1+cu121
-- Datasets 2.18.0
 - Tokenizers 0.13.3

 # SWIN-AI-Image-Detector
+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.0539
+- Accuracy: 0.9807
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.2411        | 1.0   | 547  | 0.2430          | 0.9013   |
+| 0.2011        | 2.0   | 1094 | 0.1053          | 0.9593   |
+| 0.1722        | 3.0   | 1641 | 0.0825          | 0.9671   |
+| 0.1424        | 4.0   | 2188 | 0.0851          | 0.9686   |
+| 0.1244        | 5.0   | 2735 | 0.0714          | 0.9733   |
+| 0.1089        | 6.0   | 3282 | 0.0712          | 0.9734   |
+| 0.1047        | 7.0   | 3829 | 0.0461          | 0.9833   |
+| 0.1079        | 8.0   | 4376 | 0.0454          | 0.9829   |
+| 0.0805        | 9.0   | 4923 | 0.0577          | 0.9790   |
+| 0.0778        | 10.0  | 5470 | 0.0539          | 0.9807   |
 ### Framework versions
 - Transformers 4.30.0
 - Pytorch 2.2.1+cu121
+- Datasets 2.19.0
 - Tokenizers 0.13.3