PyTorch implementation of LeNet5. Support running on single GPU. The accuracy on test set now reaches 99.9%
TODO:
- Apply Parallel training on multi-GPU
- Add the load of model from checkpoint
- TBD
-
Notifications
You must be signed in to change notification settings - Fork 0
PyTorch implementation of LeNet5. Support running on single GPU.
License
lodino/LeNet5-PyTorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
PyTorch implementation of LeNet5. Support running on single GPU.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published