added a new folder for pretrained models and updated the requirements file

2025-01-28 20:50:10 +01:00 · 2025-01-28 20:50:10 +01:00 · 9d9911b478
commit 9d9911b478
parent 7a402ea834
3 changed files with 4 additions and 33 deletions
--- a/pretrained_models/bayes_cat_dog_classifier.pth
+++ b/pretrained_models/bayes_cat_dog_classifier.pth
--- a/requirements.txt
+++ b/requirements.txt
@ -1,36 +1,5 @@
-contourpy==1.3.1
-cycler==0.12.1
-filelock==3.17.0
-fonttools==4.55.6
-fsspec==2024.12.0
-Jinja2==3.1.5
-kiwisolver==1.4.8
-MarkupSafe==3.0.2
 matplotlib==3.10.0
-mpmath==1.3.0
-networkx==3.4.2
 numpy==2.2.2
-nvidia-cublas-cu12==12.4.5.8
-nvidia-cuda-cupti-cu12==12.4.127
-nvidia-cuda-nvrtc-cu12==12.4.127
-nvidia-cuda-runtime-cu12==12.4.127
-nvidia-cudnn-cu12==9.1.0.70
-nvidia-cufft-cu12==11.2.1.3
-nvidia-curand-cu12==10.3.5.147
-nvidia-cusolver-cu12==11.6.1.9
-nvidia-cusparse-cu12==12.3.1.170
-nvidia-nccl-cu12==2.21.5
-nvidia-nvjitlink-cu12==12.4.127
-nvidia-nvtx-cu12==12.4.127
 opencv-python==4.11.0.86
-packaging==24.2
-pillow==11.1.0
-pyparsing==3.2.1
-python-dateutil==2.9.0.post0
-setuptools==75.8.0
-six==1.17.0
-sympy==1.13.1
 torch==2.5.1
 torchvision==0.20.1
-triton==3.1.0
-typing_extensions==4.12.2
--- a/train.py
+++ b/train.py
@ -9,7 +9,9 @@ import time

 IMG_SIZE = 128

-DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+#DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+DEVICE = "mps"
+

 """
 This function loads all the images from the folder and labels them