Spaces:

rzimmerdev
/

lenet_mnist

Sleeping

App Files Files Community

lenet_mnist / src

2 contributors

History: 6 commits

rzimmerdev's picture

feature: fixed training loop arguments for Lightning module

2262103 about 2 years ago

dataset.py

1.77 kB

fix: Changed default dataset image type about 2 years ago
downloader.py

1.87 kB

Finished MNIST downloading and caching modules about 2 years ago
main.py
1.29 kB

Main functionality module added: Dataset loading, caching and PyTorch Lightning calls about 2 years ago
models.py

1.6 kB

Added LeNet model implemented with PyTorch modules about 2 years ago
trainer.py

1.16 kB

feature: fixed training loop arguments for Lightning module about 2 years ago