save_hyperparameters()
is very slow when there are many hyperparameters, any speed up?
#11281
-
Hi, However, when trying a bigger version of the model (9.1 M params), the initialization of the model takes a lot more time (even before lightning prints the number of params). I've managed to nail it down to the Removing Thanks! |
Beta Was this translation helpful? Give feedback.
Replies: 4 comments 2 replies
-
What objects are you passing in? |
Beta Was this translation helpful? Give feedback.
-
Thanks for answering! I'm simply calling
|
Beta Was this translation helpful? Give feedback.
-
Awesome thanks I’ll try this out! Just to make sure, this will still allow me to checkout the best model afterward? I’m unsure what you meant by the “signature” of the model. Thanks!
… On Jan 1, 2022, at 10:33 AM, Adrian Wälchli ***@***.***> wrote:
You have to set self.save_hyperparameters(ignore="model"), since you are passing in a full model as a "hyperparameter", you don't want to save that!
—
Reply to this email directly, view it on GitHub, or unsubscribe.
You are receiving this because you authored the thread.
|
Beta Was this translation helpful? Give feedback.
-
You have to set |
Beta Was this translation helpful? Give feedback.
You have to set
self.save_hyperparameters(ignore="model")
, since you are passing in a full model as a "hyperparameter", you don't want to save that!