From 9d9911b478820f04c8db34b21fd999b5b0e6be7b Mon Sep 17 00:00:00 2001
From: Nabil Ould Hamou <ouldhamounabil@gmail.com>
Date: Tue, 28 Jan 2025 20:50:10 +0100
Subject: [PATCH] added a new folder for pretrained models and updated the
 requirements file

---
 .../bayes_cat_dog_classifier.pth              | Bin
 requirements.txt                              |  33 +-----------------
 train.py                                      |   4 ++-
 3 files changed, 4 insertions(+), 33 deletions(-)
 rename bayes_cat_dog_classifier.pth => pretrained_models/bayes_cat_dog_classifier.pth (100%)

diff --git a/bayes_cat_dog_classifier.pth b/pretrained_models/bayes_cat_dog_classifier.pth
similarity index 100%
rename from bayes_cat_dog_classifier.pth
rename to pretrained_models/bayes_cat_dog_classifier.pth
diff --git a/requirements.txt b/requirements.txt
index 7d1727d..823c81b 100644
--- 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
+torchvision==0.20.1
\ No newline at end of file
diff --git a/train.py b/train.py
index 5387bb6..4459137 100644
--- 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