Files
mlx-examples/mnist/README.md

30 lines
415 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:
2023-11-29 08:17:26 -08:00
```
python main.py --gpu
```
For a full list of options run:
```
python main.py --help
```
2023-12-24 19:49:53 +05:30
To run the PyTorch or JAX examples install the respective framework.