An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.
GNU General Public License v3.0
65
stars
9
forks
source link
Checkpointing with lightning module subclasses does not save subclass hyperparameters #119
Open
cnellington opened 2 years ago
To reproduce:
Workaround with torch
state_dict
: