My project to classify cat and dog using pytorch!
CUDA, conda installed.
conda create --name torch-cuda
conda activate torch-cuda
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
KAGGLE: Dogs vs. Cats Redux: Kernels Edition.
The train folder contains 25,000 images of dogs and cats.
The test folder contains 12,500 images of dogs and cats - non classify.
preprocess.py
Before training, create a folder name weights to save model.
train.py
predict.py