mlx-examples/mnist/README.md

30 lines
415 B
Markdown
Raw Permalink Normal View History

2023-11-30 00:17:26 +08:00
# MNIST
2023-12-12 10:45:39 +08:00
This example shows how to run some simple models on MNIST.
Install the dependencies:
```
pip install -r requirements.txt
```
2023-11-30 00: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-30 00:17:26 +08:00
```
python main.py --gpu
```
For a full list of options run:
```
python main.py --help
```
2023-12-24 22:19:53 +08:00
To run the PyTorch or JAX examples install the respective framework.