added CIFAR10 + ResNet example

This commit is contained in:
Sarthak Yadav
2023-12-12 19:01:06 +01:00
parent 13f1142eaa
commit f37e777243
5 changed files with 310 additions and 0 deletions

31
cifar/README.md Normal file
View File

@@ -0,0 +1,31 @@
# CIFAR and ResNets
* This example shows how to run ResNets on CIFAR10 dataset, in accordance with the original [paper](https://arxiv.org/abs/1512.03385).
* Also illustrates how to use `mlx-data` to download and load the dataset.
## Pre-requisites
* Install the dependencies:
```
pip install -r requirements.txt
```
## Running the example
Run the example with:
```
python main.py
```
By default the example runs on the GPU. To run on the CPU, use:
```
python main.py --cpu_only
```
For all available options, run:
```
python main.py --help
```