Skip to content

Commit 625042f

Browse files
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

.gitignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
/target
22
*.bin
3-
**/data/**
3+
**/data/**
4+
examples/testing

0 commit comments

Comments
 (0)