search

ICSM / pgmuvi

Python gaussian processes for inference on multi-wavelength light curves
https://pgmuvi.readthedocs.io
GNU General Public License v3.0
12 stars 6 forks source link

Convert Lightcurve to a subclass of torch.nn.Module? #28

Open pscicluna opened 1 year ago

pscicluna commented 1 year ago

Right now Lightcurve is a regular class. I think if it were converted to a Module, we could leverage more of torch's useful features and simplify the structure of the class a bit.

pscicluna commented 1 year ago

This looks like a good idea to make sure that GPU computing works more smoothly. both Lightcurve and Transformer need to be converted, and all their tensor attributes should be created with self.register_buffer(...)