Pytorch codes for image classification
Please prepare an environment with python=3.8, and then use the command "pip install -r requirements.txt" for the dependencies.
You may need to do the following.
conda create -n xxx python=3.8 -y
conda activate xxx
pip install -r requirements.txt
###requirements.txt
torchvision==0.14.0
torch==1.13.0
numpy==1.24.2
tensorboard==2.13.0