Skip to content
Snippets Groups Projects
Select Git revision
  • acd922f03224759db27581dc94f39c25abdbda6e
  • main default protected
2 results

mod-4-6-td-2-zhang

Forked from Dellandrea Emmanuel / MOD_4_6-TD2
4 commits ahead of the upstream repository.

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.