LSSTDESC / derp

A first attempt at a simple LSST DRP catalog emulator
BSD 3-Clause "New" or "Revised" License
1 stars 1 forks source link

Implement deep learning data I/O #6

Open jiwoncpark opened 5 years ago

jiwoncpark commented 5 years ago

We need to read in the pickled Emulator class, preprocess the data further if necessary, and create training batches. The class DerpData represents the dataset itself and will inherit from torch.utils.data.Dataset so that we can load batches by instantiating torch.utils.data.Dataloader with DerpData.