graphnet icon indicating copy to clipboard operation
graphnet copied to clipboard

Update the example scripts to use the new `GraphNeTDataModule`

Open RasmusOrsoe opened this issue 1 year ago • 2 comments

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.

RasmusOrsoe avatar Feb 27 '24 14:02 RasmusOrsoe

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.

OscarBarreraGithub avatar Jul 02 '24 16:07 OscarBarreraGithub

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

RasmusOrsoe avatar Jul 04 '24 09:07 RasmusOrsoe

Closed by #730

RasmusOrsoe avatar Sep 18 '24 13:09 RasmusOrsoe