Made ViT from scratch

This commit is contained in:
Koshin S Hegde 2024-09-09 06:53:14 +05:30
parent c856ff46f8
commit 9d0dc76903

27
src/main.py Normal file
View File

@ -0,0 +1,27 @@
import pandas
import datasets
import numpy
#|%%--%%| <66pBRkviU5|LMiIYmSiRI>
_ = datasets.load_dataset(
"Hemg/deepfake-and-real-images",
split="train"
)
if not isinstance(_, datasets.Dataset):
raise Exception("Something went wrong")
original_raw_dataset: datasets.Dataset = _
#|%%--%%| <LMiIYmSiRI|8Yvl492qqH>
_ = original_raw_dataset.to_pandas()
if not isinstance(_, datasets.Dataset):
raise Exception("Something went wrong")
original_dataset: datasets.Dataset = _
#|%%--%%| <8Yvl492qqH|SZqCHvaVss>