Image classification - Zhengfei ZHANG
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.