- Clone Repository
git clone https://github.com/MaryamBoneh/DeepLearning-Course.git
- CD Fashion-mnist-Pytorch
- pip install requirements.txt
Run train.py for start training. you should select your device(cpu is default).
like following command:
python train.py --device cuda
python test.py --device cuda
Give the input photo with the --input argument.
python inference.py --device cuda --weight fashion-mnist.pth --input test_images/shirt.png