- This repository contains the notebook in which I replicate the ViT paper from scratch in PyTorch using the AI-generated vs real images dataset which contains about 7000 labelled images which are either AI-generated or a real image.
- Here's the dataset: https://huggingface.co/datasets/mncai/Fake_or_Real_Competition_Dataset
- Medium Tutorial/Article: https://medium.com/@mohdzeesh2002/fake-vit-b-16-replicating-and-fine-tuning-the-vision-transformer-paper-from-scratch-with-pytorch-15b279fb1ee4
-
Notifications
You must be signed in to change notification settings - Fork 0
Colab notebook where I replicate the Vision Transformer research paper from scratch in PyTorch on 'AI generated vs Real images dataset' and also fine-tune a pre-trained ViT-B/16 model on the same dataset to compare its performance
zappy586/FAKE-ViT
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Colab notebook where I replicate the Vision Transformer research paper from scratch in PyTorch on 'AI generated vs Real images dataset' and also fine-tune a pre-trained ViT-B/16 model on the same dataset to compare its performance
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published