search

Lightning-AI / deep-learning-project-template

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

Replace data loaders to LightningDataModule #20

Closed 21jun closed 3 years ago

21jun commented 3 years ago

I was trying to organize datasets and dataloaders (train, val, test) in LightningDataModule