wbenoit commited on
Commit
685aa7b
·
verified ·
1 Parent(s): 9468902

Upload 2 files

Browse files
Files changed (2) hide show
  1. amplfi_hl_config.yaml +28 -0
  2. 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