Skip to content

add heterodyne functionality to aframe#440

Merged
bhgupta20 merged 5 commits into
ML4GW:mainfrom
bhgupta20:heterodyne-prototype
Apr 21, 2026
Merged

add heterodyne functionality to aframe#440
bhgupta20 merged 5 commits into
ML4GW:mainfrom
bhgupta20:heterodyne-prototype

Conversation

@bhgupta20
Copy link
Copy Markdown
Contributor

This PR adds heterodyne functionality to the train and export modules to run in the full aframe setup.

Copy link
Copy Markdown
Contributor

@wbenoit26 wbenoit26 left a comment

Choose a reason for hiding this comment

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

Mostly looks great, just a few comments.

Comment thread libs/utils/utils/preprocessing.py Outdated
chirp_mass_spacing (Literal["linear", "log"]):
Spacing of chirp mass grid. Use "linear" for evenly spaced
values or "log" for logarithmic spacing.
keep_last_n_seconds (float):
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

I think it would be more intuitive to have this be an optional parameter that defaults to None, with None meaning that the full kernel length is kept.

def build_transforms(self, *args, **kwargs):
super().build_transforms(*args, **kwargs)
self.heterodyne_transform = Heterodyne(
sample_rate=int(self.hparams.sample_rate),
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Why do these need to be cast to int? In particular, kernel_length is often a float.

Comment thread libs/utils/utils/preprocessing.py Outdated
Comment thread libs/utils/utils/preprocessing.py Outdated
Comment thread libs/utils/utils/preprocessing.py Outdated
Comment thread libs/utils/utils/preprocessing.py
@bhgupta20 bhgupta20 changed the title adding heterodyne functionality to aframe add heterodyne functionality to aframe Apr 17, 2026
@bhgupta20 bhgupta20 merged commit e290575 into ML4GW:main Apr 21, 2026
11 checks passed
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.

3 participants