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](#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](#prerequisites)
-[Description](#description)
-[Usage](#usage)
## 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.
