mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-12-16 02:08:55 +08:00
added CIFAR10 + ResNet example
This commit is contained in:
31
cifar/README.md
Normal file
31
cifar/README.md
Normal 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
|
||||
```
|
||||
Reference in New Issue
Block a user