Skip to content

[WIP] Add decimated multi-branch architectures, SVD callback, and model classes#426

Open
mcoughlin wants to merge 1 commit intoML4GW:mainfrom
mcoughlin:bns-svd-core
Open

[WIP] Add decimated multi-branch architectures, SVD callback, and model classes#426
mcoughlin wants to merge 1 commit intoML4GW:mainfrom
mcoughlin:bns-svd-core

Conversation

@mcoughlin
Copy link

Add SupervisedDecimatedResNet and SupervisedDecimatedSVDNet architectures for multi-rate decimated BNS detection. Add SVDUnfreezeCallback for two-phase training with frozen/unfrozen SVD projection layers. Add SupervisedDecimatedAframe and SimpleDecimatedAframe model classes for multi-branch training and validation.

Depends on ml4gw svd-projection branch (ml4gw.nn.svd).

…sses

Add SupervisedDecimatedResNet and SupervisedDecimatedSVDNet architectures
for multi-rate decimated BNS detection. Add SVDUnfreezeCallback for
two-phase training with frozen/unfrozen SVD projection layers. Add
SupervisedDecimatedAframe and SimpleDecimatedAframe model classes for
multi-branch training and validation.

Depends on ml4gw svd-projection branch (ml4gw.nn.svd).

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@mcoughlin mcoughlin requested a review from wbenoit26 February 16, 2026 19:49
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