Made ViT from scratch
This commit is contained in:
parent
c856ff46f8
commit
9d0dc76903
27
src/main.py
Normal file
27
src/main.py
Normal 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>
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user