search

PEREGRINE-GW / peregrine

A simulation-based Inference (SBI) library designed to perform analysis on a wide class of gravitational wave signals
MIT License
12 stars 4 forks source link

[FR] Add support for custom dataloaders #16

Open james-alvey-42 opened 8 months ago

james-alvey-42 commented 8 months ago

Is your feature request related to a problem? Please describe. Add support for custom pytorch dataloaders so that not entirely dependent on zarr stores

Describe the solution you'd like Modularise the training function so that it can take in a zarrstore or custom dataloader from the user

james-alvey-42 commented 8 months ago

Not necessarily about writing lots of custom ones ourselves, just support for any user wanting to implement their own - but need to give an example of what the dataloader needs to look like