mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-26 18:51:18 +08:00
32 lines
555 B
Markdown
32 lines
555 B
Markdown
|
# 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
|
||
|
|
```
|