Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enable param sharing with differently sized groups #514

Merged
merged 7 commits into from
Dec 5, 2024

Conversation

jnsbck
Copy link
Contributor

@jnsbck jnsbck commented Nov 19, 2024

This PR attempts to enable param sharing for controll groups with different numbers of members. This can be important for sharing params between branches with different numbers of segments or edges with different numbers of pre synaptic cells (see #501).

This also fixes #501.

todos:

  • add tests
  • rm asserts that prev. prevented the user from doing this.

@jnsbck jnsbck force-pushed the enable_inhom_param_sharing branch from aea634d to ab47228 Compare December 4, 2024 18:30
@jnsbck
Copy link
Contributor Author

jnsbck commented Dec 4, 2024

Surprisingly little had to be changed for this to work. I have added a test that checks make_trainable and set behave the same. Lemme know if any other tests are required.

Copy link
Contributor

@michaeldeistler michaeldeistler left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you, this is great!

@jnsbck
Copy link
Contributor Author

jnsbck commented Dec 5, 2024

Thanks for adding the clarifying comment :)

@jnsbck jnsbck merged commit 4ea4c16 into main Dec 5, 2024
1 check passed
@jnsbck jnsbck deleted the enable_inhom_param_sharing branch December 5, 2024 18:04
@jnsbck
Copy link
Contributor Author

jnsbck commented Dec 5, 2024

fyi @kyralianaka this has been merged now, please let me know if anything does not work as expected!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Parameter sharing with different numbers of synapses
2 participants