competitions/aiornot
Viewer • Updated • 62.1k • 50 • 32
How to use kgmann/ai-image-det-resnet18 with timm:
import timm
model = timm.create_model("hf_hub:kgmann/ai-image-det-resnet18", pretrained=True)This is a small resnet18 pretrained model, fine-tuned for 5 epochs on 80% of the AI or Not dataset and evaluated on the remaining 20% of the training dataset. It has an accuracy of 99% on the validation dataset.