Skip to content

Rework implementation of boosting for channel-slicing slimmable WaveNet, update model export version to 0.7.0.#644

Merged
sdatkinson merged 4 commits intomainfrom
slim-boost
Mar 13, 2026
Merged

Rework implementation of boosting for channel-slicing slimmable WaveNet, update model export version to 0.7.0.#644
sdatkinson merged 4 commits intomainfrom
slim-boost

Conversation

@sdatkinson
Copy link
Owner

Code feels cleaner. I could've sworn that I saw nonzero grads where they shouldn't have been as well, but apparently the reworked test still passes on main, so 🤷🏻

- SlimmableConv1dBase._get_slim_indices derives i_in, i_out from ratio
  and allowed channel lengths (configured in each subclass __init__)
- Remove redundant _get_slim_indices overrides from RechannelIn,
  LayerConv, InputMixer, Layer1x1, HeadRechannel
- Add _is_first default to base; remove _output_paired, abc import
- Head1x1 retains override (NotImplementedError)

Made-with: Cursor
@sdatkinson
Copy link
Owner Author

Golly, the graceful shutdown tests are flaky

@sdatkinson sdatkinson merged commit 755380a into main Mar 13, 2026
8 of 9 checks passed
@sdatkinson sdatkinson deleted the slim-boost branch March 13, 2026 23:32
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.

1 participant