Commit 625042f
committed
Refactor MNIST example and enhance training options
- Updated the MNIST GPU example to include TrainOptions for better training configuration.
- Simplified the loading of MNIST data and ensured proper handling of existing files.
- Improved the neural network architecture in the MNIST example for better performance.
- Added a new drawing example for MNIST digit classification using eframe and egui.
- Moved serialization functionality for saving and loading neural network models.
- Enhanced the training module with a callback system for monitoring training progress.
- Cleaned up the backend module to ensure only one backend feature can be active at a time.
- Fixed minor issues in the transpose shader comments and ensured consistent formatting.1 parent 2e6f73b commit 625042f
14 files changed
Lines changed: 4419 additions & 379 deletions
File tree
- examples/mnist
- src
- backend
- gpu/shaders
- metric
- optimizer
- serialization
- train
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
3 | | - | |
| 3 | + | |
| 4 | + | |
0 commit comments