graphnet-team / graphnet

A Deep learning library for neutrino telescopes
https://graphnet-team.github.io/graphnet/
Apache License 2.0
85 stars 86 forks source link

Adopted pytorch lightning DataModule instead of DataLoaders and Dataset #642

Closed samadpls closed 5 months ago

samadpls commented 7 months ago

Adopted PyTorch Lightning DataModule instead of managing individual DataLoaders and Dataset. This pull request addresses and resolves issue #579

samadpls commented 5 months ago

@RasmusOrsoe, thank you for reviewing and giving your feedback. I have implemented the necessary adjustments to address the comment. 😋