Skip to content

Added MNIST MLP benchmark#75

Merged
ClashLuke merged 6 commits into
HomebrewML:mainfrom
tom-jod:main
Sep 19, 2025
Merged

Added MNIST MLP benchmark#75
ClashLuke merged 6 commits into
HomebrewML:mainfrom
tom-jod:main

Conversation

@tom-jod

@tom-jod tom-jod commented Sep 9, 2025

Copy link
Copy Markdown
Contributor

and edited the utils script to report the test accuracies every {group} number of steps.

Simple, fast to run real world benchmarks using minibatches of real data would be a useful addition to heavyball to help validate the correlation between the toy problems and the real world uses cases that we care about optimizing.

Future PRs will add more of these small scale benchmarks

@ClashLuke

Copy link
Copy Markdown
Member

lgtm

@ClashLuke ClashLuke merged commit a897a58 into HomebrewML:main Sep 19, 2025
1 check failed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants