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