Update amplfi-hl-config.yaml

amplfi-hl-config.yaml

@@ -17,7 +17,90 @@ architecture:
           init_args:
             groups: 8
 parameter_sampler: 
-    class_path: buoy.utils.pe.parameter_sampler
+  class_path: amplfi.train.prior.AmplfiPrior
+  init_args:
+    conversion_function: amplfi.train.conversion.precessing_to_lalsimulation_parameters
+    priors:
+      chirp_mass: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 10
+          high: 100
+          validate_args: false
+      mass_ratio: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.125 
+          high: 0.999
+          validate_args: false
+      distance: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low:  100 
+          high: 3100 
+          validate_args: false
+      inclination: 
+        class_path: ml4gw.distributions.Sine
+        init_args:
+          low: 0 
+          high: 3.14159 
+          validate_args: false
+      phic: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.0 
+          high: 6.28318 
+          validate_args: false
+      a_1: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.0 
+          high: 0.999
+          validate_args: false
+      a_2: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.0 
+          high: 0.999
+          validate_args: false
+      tilt_1: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.0 
+          high: 3.14159  
+          validate_args: false
+      tilt_2: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.0 
+          high: 3.14159 
+          validate_args: false
+      phi_jl: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.0 
+          high: 6.28318 
+          validate_args: false
+      phi_12: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0.0
+          high: 6.28318 
+          validate_args: false 
+      dec: 
+        class_path: ml4gw.distributions.Cosine
+      psi: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0
+          high: 3.14159
+          validate_args: false
+      phi: 
+        class_path: torch.distributions.Uniform
+        init_args:
+          low: 0
+          high: 6.28318
+          validate_args: false
 sample_rate: 2048
 inference_params: ["chirp_mass", "mass_ratio", "distance", "phic",  "inclination", "dec", "psi", "phi"]
 psd_length: 10