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

add MNIST training example

Browse Source
This commit is contained in:
David Koski
2024-03-01 15:55:36 -08:00
parent 2157333905
commit 79e0620891
11 changed files with 576 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
Applications/MNISTTrainer/README.md Normal file
View File

@@ -0,0 +1,13 @@
# MNISTTrainer
This is an example showing how to do model training on both macOS and iOS.
This will download the MNIST training data, create a new models and train
it. It will show the timing per epoch and the test accuracy as it trains.
You will need to set the Team on the MNISTTrainer target in order to build and
run on iOS.
Some notes about the setup:
- this will download test data over the network so MNISTTrainer -> Signing & Capabilities has the "Outgoing Connections (Client)" set in the App Sandbox
- the website it connects to uses http rather than https so it has a "App Transport Security Settings" in the Info.plist