.gitignore

-*.jpg
 .DS_Store
 
 # Data
 data/*
-transfer_learning/hymenoptera_data/*
+hymenoptera_data/*
 
 # Torch model
 *.pt

README.md

+# Image classification - Zhengfei ZHANG
+
+## 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.
\ No newline at end of file