Files
mlx-examples/mnist/README.md

24 lines
350 B
Markdown
Raw Normal View History

2023-11-29 08:17:26 -08:00
# MNIST
2023-12-11 20:45:39 -06:00
This example shows how to run some simple models on MNIST.
Install the dependencies:
```
pip install -r requirements.txt
```
2023-11-29 08:17:26 -08:00
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.