save new LR with save_hyperparameters after lr_find #15108
Unanswered
Shualdon
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hi,
I'm tuning the initial LR using
lr_find
, and I want to save the new LR hyperparameter usingsave_hyeperparameters
. Right now I have thesave_hyeperparameters
in the init function with some default LR I pass to the model. I do see it saves the passed value to the hparam file, but it doesn't update the value after I find a better value. I usemodel.hparams.lr = lr_finder.suggestion()
to update the LR in the model, as well asupdate_attr=True
in thelr_tune
function - just for good measure...Is there a way to save the new value? Calling
save_hyeperparameters
outside of the init function gives an error.Beta Was this translation helpful? Give feedback.
All reactions