kgmann commited on
Commit
92f73c2
1 Parent(s): f8c1043

Update README.md

Browse files

This 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.

Files changed (1) hide show
  1. README.md +4 -0
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