mmanikanta
/

ConvNeXT_AI_image_detector

Image Classification

Generated from Trainer

Model card Files Files and versions

Metrics Training metrics Community

mmanikanta commited on Apr 20, 2024

Commit

be72287

·

1 Parent(s): a48fcbb

update model card README.md

Files changed (1) hide show

README.md +13 -13

README.md CHANGED Viewed

@@ -16,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [facebook/convnext-tiny-224](https://huggingface.co/facebook/convnext-tiny-224) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0466
-- Accuracy: 0.9847
 ## Model description
@@ -51,21 +51,21 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step  | Validation Loss | Accuracy |
 |:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 0.1893        | 1.0   | 1406  | 0.1578          | 0.9399   |
-| 0.1777        | 2.0   | 2812  | 0.1009          | 0.9632   |
-| 0.1255        | 3.0   | 4218  | 0.0752          | 0.9709   |
-| 0.1296        | 4.0   | 5625  | 0.0808          | 0.9715   |
-| 0.1154        | 5.0   | 7031  | 0.1031          | 0.963    |
-| 0.089         | 6.0   | 8437  | 0.0569          | 0.9794   |
-| 0.0949        | 7.0   | 9843  | 0.0468          | 0.9846   |
-| 0.0808        | 8.0   | 11250 | 0.0466          | 0.9847   |
-| 0.1091        | 9.0   | 12656 | 0.0507          | 0.9826   |
-| 0.0765        | 10.0  | 14060 | 0.0521          | 0.9829   |
 ### Framework versions
 - Transformers 4.30.0
 - Pytorch 2.2.1+cu121
-- Datasets 2.18.0
 - Tokenizers 0.13.3

 This model is a fine-tuned version of [facebook/convnext-tiny-224](https://huggingface.co/facebook/convnext-tiny-224) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.0512
+- Accuracy: 0.9817
 ## Model description
 | Training Loss | Epoch | Step  | Validation Loss | Accuracy |
 |:-------------:|:-----:|:-----:|:---------------:|:--------:|
+| 0.2164        | 1.0   | 1093  | 0.1476          | 0.9444   |
+| 0.1647        | 2.0   | 2187  | 0.0872          | 0.9673   |
+| 0.1784        | 3.0   | 3281  | 0.0761          | 0.9709   |
+| 0.1247        | 4.0   | 4375  | 0.0869          | 0.9666   |
+| 0.0994        | 5.0   | 5468  | 0.0808          | 0.9696   |
+| 0.073         | 6.0   | 6562  | 0.0514          | 0.9820   |
+| 0.0924        | 7.0   | 7656  | 0.0526          | 0.9809   |
+| 0.0779        | 8.0   | 8750  | 0.0518          | 0.9819   |
+| 0.0903        | 9.0   | 9843  | 0.0518          | 0.9826   |
+| 0.0721        | 9.99  | 10930 | 0.0512          | 0.9817   |
 ### Framework versions
 - Transformers 4.30.0
 - Pytorch 2.2.1+cu121
+- Datasets 2.19.0
 - Tokenizers 0.13.3