cnellington / Contextualized

An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.
http://contextualized.ml/
GNU General Public License v3.0
65 stars 9 forks source link

Move models to PyTorch Lightning #47

Closed cnellington closed 2 years ago

cnellington commented 2 years ago

resolves #4 resolves #18 resolves #46 (for future: NOTMAD still needs to be converted to torch) progress on #3 (should be possible now by subclassing the lightning trainer + dataloader to do batch-wise dimensionality reduction on networks)