Hugging Face's logo

Spaces:
rzimmerdev
/
lenet_mnist
Sleeping

App Files Community
lenet_mnist / src

Commit History

Fixed devices and demo
a0f925f

rzimmerdev commited on

feature: Added prediction and model loading methods
abe84f3

rzimmerdev commited on

feature: Added manual training and PyTorch Lightning training loops
987f571

rzimmerdev commited on

fix: Removed last activation layer from LeNet model
7dc7452

rzimmerdev commited on

feature: fixed training loop arguments for Lightning module
2262103

rzimmerdev commited on

fix: Changed default dataset image type
1de9461

rzimmerdev commited on

Main functionality module added: Dataset loading, caching and PyTorch Lightning calls
49b098d

rzimmerdev commited on

Implemented training loops and logging functionalities with PyTorch Lightning
b358118

rzimmerdev commited on

Added LeNet model implemented with PyTorch modules
82fdb01

rzimmerdev commited on

Finished MNIST downloading and caching modules
6ec3bf6

rzimmerdev commited on