Upgrade to lightning and pytorch 2

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

Upgrade to lightning and pytorch 2 #181

Closed cnellington closed 4 months ago

cnellington commented 1 year ago

upgrade to torch 2.0+ and use .compile() where possible change pytorch-lightning dependency to lightning and refactor all import pytorch_lightning statements to import lightning.pytorch

cnellington commented 1 year ago

Linking #201 for pytorch upgrade