Spaces:

rzimmerdev
/

lenet_mnist

Sleeping

App Files Files Community

lenet_mnist / src

2 contributors

History: 9 commits

rzimmerdev's picture

feature: Added prediction and model loading methods

abe84f3 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
models.py

1.67 kB

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

1.49 kB

feature: Added prediction and model loading methods about 2 years ago
train.py

1.89 kB

feature: Added manual training and PyTorch Lightning training loops about 2 years ago
trainer.py

1.16 kB

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