mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-01 12:29:42 +08:00
a few examples
This commit is contained in:
18
mnist/README.md
Normal file
18
mnist/README.md
Normal file
@@ -0,0 +1,18 @@
|
||||
# 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.
|
Reference in New Issue
Block a user