Pre-requisites:

python(3.8.18)
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
store datasets in the "dataset" folder.
run "get_indexes.py" to get the indices of train and test files for each of the folds.
run "main.py"