Skip to content
Discussion options

You must be logged in to vote

When you have nested LightningModules, their configure_optimizers will never be called unless you explicitly call it in the top-level configure_optimizers.

That being said, if you call, merge and return the optimizers created there, these optimizers should only contain parameters from the respective submodule (if implemented correctly)

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by Co1lin
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment