feat(pt): implement shared_dict functionality for linear models #4933
+381
−1
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR implements
shared_dict
parameter support forLinearEnergyModel
to enable parameter sharing between submodels, following the same pattern used in multi-task models.Problem
Users requested the ability to share parameters between submodels in linear models, similar to how multi-task models work. The original linear model implementation only supported independent submodels without parameter sharing capabilities.
Solution
Added support for the
shared_dict
parameter toLinearEnergyModel
by:preprocess_shared_params()
to convert shared references to actual parameter dictionaries_share_params()
method adapted fromModelWrapper.share_params()
to enable actual parameter sharing between submodelsUsage Example
The implementation reuses existing multi-task infrastructure to minimize code changes while providing the requested functionality.
Fixes #4412.
✨ Let Copilot coding agent set things up for you — coding agent works faster and does higher quality work when set up for your repo.