search

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

Update the example scripts to use the new `GraphNeTDataModule` #669

Open RasmusOrsoe opened 4 months ago

RasmusOrsoe commented 4 months ago

Since #642 we've gotten a much more elegant way of producing dataloaders for training and inference. We should update our training scripts to use this new method instead, and strongly consider to remove the old method, as discussed in #467.

OscarBarreraGithub commented 2 weeks ago

I got this to run locally on my machine by passing the config dict variables into the GraphNetDataModule arguments and calling the train_dataloader and val_dataloader properties. This implementation works with both .db files and parquet paths containing separate directories for pulse maps and truth tables. I will submit a pull request soon.

RasmusOrsoe commented 2 weeks ago

@OscarBarreraGithub Sounds great! I assigned you to the issue.