The example errors out quickly, likely because we are not properly initializing a lens using the stored state_dict in our checkpoints, we also need to make sure that we use the LightningLens class to consume these parameters, because that is the object whose state_dict we are saving.
The example errors out quickly, likely because we are not properly initializing a lens using the stored
state_dictin our checkpoints, we also need to make sure that we use theLightningLensclass to consume these parameters, because that is the object whosestate_dictwe are saving.