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.

Sign up for GitHub

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

Jump to bottom

Integrate BNS model training code #125

Closed

wbenoit26 opened this issue Apr 9, 2024 · 1 comment · Fixed by #138

Assignees

Contributor

wbenoit26 commented Apr 9, 2024

No description provided.

wbenoit26 self-assigned this

wbenoit26 mentioned this issue

Add BNS prior ML4GW/aframe-legacy#464

Closed

Contributor

EthanMarx commented Apr 15, 2024

I'll point you to the frequency domain dataset. This is where q transform / spectrogram augmentations would go. Also, the [frequency domain arch] (https://github.com/ML4GW/aframev2/blob/main/projects/train/train/architectures/supervised.py#L91) is a wrapper around ResNet2d that plays nicely with lightning.

Changing the model arch class path and dataset class path in the train config should get you started.

wbenoit26 mentioned this issue

Merged

wbenoit26 closed this as completed in #138

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment