Image classification - Zhengfei ZHANG
This project consists of my solution for the image classification assignment.
If you need to run the code, I suggest you doing it first (see Usage), then read the rest of this file for details and explanations while the program is running, as it is rather time consuming.
Table of Contents
Prerequisites
All of the work has been performed under Python 3.12.2.
The libraries and the version used are as follows:
- matplotlib 3.9.2
- numpy 2.0.2
- pandas 2.2.2
- PyTorch 2.5.1
- Torchvision 0.20.1
Description
The goal of the project is to do image classification on the CIFAR-10 dataset using deep learning methods.
Using macOS with Apple Silicon
For the quantization part, mps isn't supported yet. I had to move it to CPU for this one. It is significantly slower.