Model training and evaluation code for Hugging Face AI or Not Competition. Primary objective is to achieve minimal cross entropy loss for binary image classiciation task of determining whether an image is AI generated or not. All model finetuning is performed on a single GPU through Colab. The competition dataset is used to finetune VIT-base, VIT-large, and Facebook's VIT mae (large and base) transformers. Best log loss achieved:
Code for Hugging Face AI or Not Competition
sunil-2000/aiornot
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Code for Hugging Face AI or Not Competition
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published