zhangshuai/mlx-swift-examples
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

LeNet on MNIST + readme update (#12)

Browse Source 
* LeNet on MNIST + readme update

* tanh + remove device toggle

* remove device entirely
This commit is contained in:
Awni Hannun
2024-03-04 14:16:20 -08:00
committed by GitHub
parent dfc9f2fc01
commit 4ed4ec69e7
8 changed files with 56 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
Libraries/MNIST/README.md
View File

@@ -1,13 +1,11 @@
# MNIST
This is a port of the MNIST model and training code from:
- https://github.com/ml-explore/mlx-examples/blob/main/mnist
This is a port of the MNIST training code from the [Python MLX example](https://github.com/ml-explore/mlx-examples/blob/main/mnist). This example uses a [LeNet](https://en.wikipedia.org/wiki/LeNet) instead of an MLP.
It provides code to:
- download the test/train data
- provides the MNIST model (MLP)
- some functions to shuffle and batch the data
- Download the MNIST test/train data
- Build the LeNet
- Some functions to shuffle and batch the data
See [mnist-tool](../../Tools/mnist-tool) for an example of how to run this. The training loop also lives there.
See [mnist-tool](../../Tools/mnist-tool) for an example of how to run this. The training loop also lives there.