Update README.md
Browse filesThis is a small **resnet18** pretrained model, fine-tuned for 5 epochs on 80% of the [AI or Not dataset](https://huggingface.co/datasets/competitions/aiornot) and evaluated on the remaining 20% of the training dataset.
It has an accuracy of 85% on the validation dataset.
README.md
CHANGED
|
@@ -3,5 +3,9 @@ tags:
|
|
| 3 |
- image-classification
|
| 4 |
- timm
|
| 5 |
library_tag: timm
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
---
|
| 7 |
# Model card for kgmann/ai-image-det-resnet18
|
|
|
|
| 3 |
- image-classification
|
| 4 |
- timm
|
| 5 |
library_tag: timm
|
| 6 |
+
datasets:
|
| 7 |
+
- competitions/aiornot
|
| 8 |
+
metrics:
|
| 9 |
+
- accuracy
|
| 10 |
---
|
| 11 |
# Model card for kgmann/ai-image-det-resnet18
|