Spaces:

rzimmerdev
/

lenet_mnist

Running

App Files Files Community

lenet_mnist / src

7.69 kB

2 contributors

History: 5 commits

rzimmerdev's picture

fix: Changed default dataset image type

1de9461 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
main.py

1.29 kB

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

1.6 kB

Added LeNet model implemented with PyTorch modules almost 3 years ago
trainer.py

1.16 kB

Implemented training loops and logging functionalities with PyTorch Lightning almost 3 years ago