search

Lightning-AI / deep-learning-project-template

Pytorch Lightning code guideline for conferences
Apache License 2.0
1.23k stars 269 forks source link

Adding a LightningDataModule example for lit_minist #31

Open knoriy opened 2 years ago

knoriy commented 2 years ago

I've modified lit_minits.py to use LightningDataModule