nabil_oh/CnD_Classification
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Nabil Ould Hamou 2025-01-28 20:50:10 +01:00
parent 7a402ea834
commit 9d9911b478
3 changed files with 4 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
bayes_cat_dog_classifier.pth → pretrained_models/bayes_cat_dog_classifier.pth
View File

33
requirements.txt
View File

@ -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 matplotlib==3.10.0
mpmath==1.3.0
networkx==3.4.2
numpy==2.2.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 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 torch==2.5.1
torchvision==0.20.1 torchvision==0.20.1
triton==3.1.0
typing_extensions==4.12.2

4
train.py
View File

@ -9,7 +9,9 @@ import time
IMG_SIZE = 128 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 This function loads all the images from the folder and labels them