From 9d0dc769036f236830bb104c45d012953717cf9c Mon Sep 17 00:00:00 2001 From: kosh Date: Mon, 9 Sep 2024 06:53:14 +0530 Subject: [PATCH] Made ViT from scratch --- src/main.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 src/main.py diff --git a/src/main.py b/src/main.py new file mode 100644 index 0000000..84c5cb6 --- /dev/null +++ b/src/main.py @@ -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 = _ + +#|%%--%%| + +_ = original_raw_dataset.to_pandas() + +if not isinstance(_, datasets.Dataset): + raise Exception("Something went wrong") +original_dataset: datasets.Dataset = _ + +#|%%--%%| <8Yvl492qqH|SZqCHvaVss> + +