You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi,
I use the on_fit_start hook inside the LightningModule to freeze some parameters of my model.
The ModelSummary callback uses the the same hook. Unfortunately the hook from the callback gets called before the hook from the module, so the number of trainable and non-trainable parameters is not correct.
Is there I way I can specify the order of same hooks get called?
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
Uh oh!
There was an error while loading. Please reload this page.
-
Hi,
I use the
on_fit_start
hook inside theLightningModule
to freeze some parameters of my model.The
ModelSummary
callback uses the the same hook. Unfortunately the hook from the callback gets called before the hook from the module, so the number of trainable and non-trainable parameters is not correct.Is there I way I can specify the order of same hooks get called?
Thanks
Beta Was this translation helpful? Give feedback.
All reactions