LeNet on MNIST + readme update (#12)

* 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

View File

@@ -1,37 +1,20 @@
# MLX Swift Examples
Example [mlx-swift](https://github.com/ml-explore/mlx-swift) programs.
Example [MLX Swift](https://github.com/ml-explore/mlx-swift) programs.
## MNISTTrainer
- [MNISTTrainer](Applications/MNISTTrainer/README.md): An example that runs on
both iOS and macOS that downloads MNIST training data and trains a
[LeNet](https://en.wikipedia.org/wiki/LeNet).
An example that runs on both iOS and macOS that downloads MNIST training
data and trains an MNIST model.
- [LLMEval](Applications/LLMEval/README.md): An example that runs on both iOS
and macOS that downloads an LLM and tokenizer from Hugging Face and and
generates text from a given prompt.
- [README](Applications/MNISTTrainer/README.md)
- [LinearModelTraining](Tools/LinearModelTraining/README.md): An example that
trains a simple linear model.
## LLMEval
An example that runs on both iOS and macOS that downloads a LLM model
weights and tokenizer configuration from Hugging Face and evaluates
a prompt in-process.
- [README](Applications/LLMEval/README.md)
## LinearModelTraining
A simple linear model and a training loop.
- [README](Tools/LinearModelTraining/README.md)
## llm-tool
A command line tool for generating text using a variety of Hugging Face models:
- [README](Tools/llm-tool/README.md)
## mnist-tool
A command line tool for training an MNIST (MLP) model:
- [README](Tools/mnist-tool/README.md)
- [llm-tool](Tools/llm-tool/README.md): A command line tool for generating text
using a variety of LLMs available on the Hugging Face hub.
- [mnist-tool](Tools/mnist-tool/README.md): A command line tool for training a
a LeNet on MNIST.