Spaces:

rzimmerdev
/

lenet_mnist

Sleeping

App Files Files Community

lenet_mnist / src

2 contributors

History: 7 commits

rzimmerdev's picture

fix: Removed last activation layer from LeNet model

7dc7452 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.67 kB

fix: Removed last activation layer from LeNet model about 2 years ago
trainer.py

1.16 kB

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