mmanikanta commited on
Commit
5b4602c
1 Parent(s): d72f8f9

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -14
README.md CHANGED
@@ -14,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # SWIN-AI-Image-Detector
16
 
17
- 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.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.0342
20
- - Accuracy: 0.9874
21
 
22
  ## Model description
23
 
@@ -51,21 +51,21 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
- | 0.2432 | 1.0 | 703 | 0.1859 | 0.9267 |
55
- | 0.1665 | 2.0 | 1406 | 0.1229 | 0.9514 |
56
- | 0.1427 | 3.0 | 2109 | 0.0604 | 0.9761 |
57
- | 0.143 | 4.0 | 2813 | 0.0807 | 0.97 |
58
- | 0.1008 | 5.0 | 3516 | 0.0444 | 0.9824 |
59
- | 0.1189 | 6.0 | 4219 | 0.0635 | 0.977 |
60
- | 0.0936 | 7.0 | 4922 | 0.0535 | 0.9789 |
61
- | 0.1201 | 8.0 | 5626 | 0.0376 | 0.9854 |
62
- | 0.076 | 9.0 | 6329 | 0.0342 | 0.9874 |
63
- | 0.0887 | 10.0 | 7030 | 0.0413 | 0.9844 |
64
 
65
 
66
  ### Framework versions
67
 
68
  - Transformers 4.30.0
69
  - Pytorch 2.2.1+cu121
70
- - Datasets 2.18.0
71
  - Tokenizers 0.13.3
 
14
 
15
  # SWIN-AI-Image-Detector
16
 
17
+ 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.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.0539
20
+ - Accuracy: 0.9807
21
 
22
  ## Model description
23
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
+ | 0.2411 | 1.0 | 547 | 0.2430 | 0.9013 |
55
+ | 0.2011 | 2.0 | 1094 | 0.1053 | 0.9593 |
56
+ | 0.1722 | 3.0 | 1641 | 0.0825 | 0.9671 |
57
+ | 0.1424 | 4.0 | 2188 | 0.0851 | 0.9686 |
58
+ | 0.1244 | 5.0 | 2735 | 0.0714 | 0.9733 |
59
+ | 0.1089 | 6.0 | 3282 | 0.0712 | 0.9734 |
60
+ | 0.1047 | 7.0 | 3829 | 0.0461 | 0.9833 |
61
+ | 0.1079 | 8.0 | 4376 | 0.0454 | 0.9829 |
62
+ | 0.0805 | 9.0 | 4923 | 0.0577 | 0.9790 |
63
+ | 0.0778 | 10.0 | 5470 | 0.0539 | 0.9807 |
64
 
65
 
66
  ### Framework versions
67
 
68
  - Transformers 4.30.0
69
  - Pytorch 2.2.1+cu121
70
+ - Datasets 2.19.0
71
  - Tokenizers 0.13.3