lenet_mnist / notebooks

Commit History

Fixed devices and demo
a0f925f

rzimmerdev commited on

feature: Added prediction and model loading methods
abe84f3

rzimmerdev commited on

feature: fixed training loop arguments for Lightning module
2262103

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

Boilerplate for Trainer and Model classes
0e42d3f

rzimmerdev commited on

Created dataloader class
0e65489

rzimmerdev commited on

Added boilerplate project structure
ae1ea8a

rzimmerdev commited on