Upload 2 files
Browse files- amplfi_hl_config.yaml +28 -0
- amplfi_hlv_config.yaml +28 -0
amplfi_hl_config.yaml
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
architecture:
|
| 2 |
+
class_path: amplfi.train.architectures.flows.NSF
|
| 3 |
+
init_args:
|
| 4 |
+
hidden_features: [512, 512, 512]
|
| 5 |
+
transforms: 20
|
| 6 |
+
randperm: true
|
| 7 |
+
embedding_net:
|
| 8 |
+
class_path: amplfi.train.architectures.embeddings.MultiModalPsd
|
| 9 |
+
init_args:
|
| 10 |
+
num_ifos: 2
|
| 11 |
+
time_context_dim: 12
|
| 12 |
+
freq_context_dim: 32
|
| 13 |
+
time_layers: [5, 3, 3]
|
| 14 |
+
freq_layers: [5, 3, 3]
|
| 15 |
+
norm_layer:
|
| 16 |
+
class_path: ml4gw.nn.norm.GroupNorm1DGetter
|
| 17 |
+
init_args:
|
| 18 |
+
groups: 8
|
| 19 |
+
amplfi_parameter_sampler:
|
| 20 |
+
class_path: buoy.utils.pe.parameter_sampler
|
| 21 |
+
sample_rate: 2048
|
| 22 |
+
inference_params: ["chirp_mass", "mass_ratio", "distance", "phic", "inclination", "dec", "psi", "phi"]
|
| 23 |
+
psd_length: 10
|
| 24 |
+
fduration: 2
|
| 25 |
+
kernel_length: 3
|
| 26 |
+
event_position: 2.5
|
| 27 |
+
fftlength: 2.5
|
| 28 |
+
highpass: 25
|
amplfi_hlv_config.yaml
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
amplfi_hlv_architecture:
|
| 2 |
+
class_path: amplfi.train.architectures.flows.NSF
|
| 3 |
+
init_args:
|
| 4 |
+
hidden_features: [512, 512, 512, 512, 512]
|
| 5 |
+
transforms: 20
|
| 6 |
+
randperm: true
|
| 7 |
+
embedding_net:
|
| 8 |
+
class_path: amplfi.train.architectures.embeddings.MultiModalPsd
|
| 9 |
+
init_args:
|
| 10 |
+
num_ifos: 3
|
| 11 |
+
time_context_dim: 20
|
| 12 |
+
freq_context_dim: 48
|
| 13 |
+
time_layers: [5, 3, 3]
|
| 14 |
+
freq_layers: [5, 3, 3]
|
| 15 |
+
norm_layer:
|
| 16 |
+
class_path: ml4gw.nn.norm.GroupNorm1DGetter
|
| 17 |
+
init_args:
|
| 18 |
+
groups: 8
|
| 19 |
+
amplfi_parameter_sampler:
|
| 20 |
+
class_path: buoy.utils.pe.parameter_sampler
|
| 21 |
+
sample_rate: 2048
|
| 22 |
+
inference_params: ["chirp_mass", "mass_ratio", "distance", "phic", "inclination", "dec", "psi", "phi"]
|
| 23 |
+
psd_length: 10
|
| 24 |
+
fduration: 2
|
| 25 |
+
kernel_length: 3
|
| 26 |
+
event_position: 2.5
|
| 27 |
+
fftlength: 2.5
|
| 28 |
+
highpass: 25
|