Spaces:

rzimmerdev
/

lenet_mnist

Sleeping

App Files Files Community

lenet_mnist / src

9.84 kB

2 contributors

History: 9 commits

rzimmerdev's picture

feature: Added prediction and model loading methods

abe84f3 almost 3 years ago

dataset.py

1.77 kB

fix: Changed default dataset image type almost 3 years ago
downloader.py

1.87 kB

Finished MNIST downloading and caching modules almost 3 years ago
models.py

1.67 kB

fix: Removed last activation layer from LeNet model almost 3 years ago
predict.py

1.49 kB

feature: Added prediction and model loading methods almost 3 years ago
train.py

1.89 kB

feature: Added manual training and PyTorch Lightning training loops almost 3 years ago
trainer.py

1.16 kB

feature: fixed training loop arguments for Lightning module almost 3 years ago