mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-26 18:51:18 +08:00
19 lines
310 B
Markdown
19 lines
310 B
Markdown
|
# MNIST
|
||
|
|
|
||
|
|
This example shows how to run some simple models on MNIST. The only
|
||
|
|
dependency is MLX.
|
||
|
|
|
||
|
|
Run the example with:
|
||
|
|
|
||
|
|
```
|
||
|
|
python main.py
|
||
|
|
```
|
||
|
|
|
||
|
|
By default the example runs on the CPU. To run on the GPU, use:
|
||
|
|
|
||
|
|
```
|
||
|
|
python main.py --gpu
|
||
|
|
```
|
||
|
|
|
||
|
|
To run the PyTorch or Jax examples install the respective framework.
|