Skip to content
Snippets Groups Projects
Commit 36781999 authored by Zhang Zhengfei's avatar Zhang Zhengfei
Browse files

finished exercise 2, started exercise 3

parent 9f0b9e8b
Branches
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
README.md
+ 6
− 1
View file @ 36781999
...@@ -18,9 +18,14 @@ The libraries and the version used are as follows: ...@@ -18,9 +18,14 @@ The libraries and the version used are as follows:
- matplotlib 3.9.2 - matplotlib 3.9.2
- numpy 2.0.2 - numpy 2.0.2
- pandas 2.2.2
- PyTorch 2.5.1 - PyTorch 2.5.1
- Torchvision 0.20.1 - Torchvision 0.20.1
## Description ## Description
The goal of the project is to do image classification on the CIFAR-10 dataset using deep learning methods. 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
TD2 Deep Learning.ipynb
+ 491
− 49
View file @ 36781999
Source diff could not be displayed: it is too large. Options to address this: view the blob.
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment