search

ML4GW / amplfi

Performing parameter estimation on gravitational wave data with machine learning
MIT License
7 stars 8 forks source link

Add tuning script! #134

Closed EthanMarx closed 2 weeks ago

EthanMarx commented 3 weeks ago

  1. Add tuning script built entirely on lightray that should work for both the Similarity and Flow LightningCLI classes. If a ray cluster is active, can pass the ip to the address variable. Otherwise, will tune locally

  2. Adds amplfi_init.py script for easily initializing fresh run directories with config files for training or tuning.

  3. Adds omegaconf parser mode in AmplfiBaseCLI __init__ so the cli is compatible with tuning script

alecgunny commented 3 weeks ago

@EthanMarx Just chiming in to say I hadn't seen your lightray library and this whole thing is extremely sick

EthanMarx commented 3 weeks ago

@alecgunny Thank you thank you finally got around to generalizing things and turned out nicely!

EthanMarx commented 3 weeks ago

@deepchatterjeeligo Adds tuning script, and helper script (similar to what we have with aframe) for making initialization of run directories simpler. Will add some documentation on this.

EthanMarx commented 2 weeks ago

@deepchatterjeeligo should be good to go