mlx-examples/mnist/README.md

24 lines
350 B
Markdown
Raw 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:
```
python main.py --gpu
```
To run the PyTorch or Jax examples install the respective framework.