{ "duration": 1.0, "__definition__duration": "length of training/validation/test sample time series in seconds", "r": 10, "__definition__r": "number of GW timeseries to use for testing.", "det": ["H1","L1","V1"], "__definition__det": "LIGO detectors to perform analysis on (default is 3detector H1,L1,V1)", "z_dimension": 4, "__definition__z_dimension": "number of latent space dimensions of model", "tot_dataset_size": 50000, "__definition__tot_dataset_size": "total number of training samples available to use", "tset_split": 100, "__definition__tset_split": "number of training samples in each training data file", "val_dataset_size": 10000, "__definition__val_dataset_size": "total number of validation samples to use", "ramp": true, "__definition__ramp": "# if true, apply linear ramp to KL loss", "print_values": true, "__definition__print_values": "# optionally print loss values every report interval", "load_plot_data": false, "__definition__load_plot_data": "Use plotting data which has already been generated", "doPE": false, "__definition__doPE": "if True then do bilby PE when generating new testing samples", "gpu_num": 0, "__definition__gpu_num": "gpu number run is running on", "ndata": 262142, "__definition__ndata": "sampling frequency", "run_label": "chris_tiny_3detbench+tf2.3_nopool_model", "__definition__run_label": "label of run", "bilby_results_label": "all_1_samplers", "__definition__bilby_result_label": "label given to bilby results directory", "plot_dir": "/data/www.astro/2263373r/plots", "__definition__plot_dir": "output directory to save results plots", "load_chunk_size": 250, "__definition__load_chunk_size": "Number of training samples to load in at a time.", "load_iteration": 5000, "__definition__load_iteration": "How often to load another chunk of training samples", "weight_init": "xavier", "__definition__weight_init": "[xavier,VarianceScaling,Orthogonal]. Network model weight initialization (default is xavier)", "n_samples": 10000, "__definition__n_samples": "number of posterior samples to save per reconstruction upon inference (default 3000)", "num_iterations": 10000001, "__definition__num_iterations": "total number of iterations before ending training of model", "initial_training_rate": 0.00004, "__definition__initial_training_rate": "initial training rate for ADAM optimiser inference model (inverse reconstruction)", "extra_lr_decay_factor": false, "__definition__extra_decay_factor": "Use an extra decay factor of 0.95 after every load iteration", "batch_size": 512, "__definition__batch_size": "Number training samples shown to neural network per iteration", "n_KLzsamp": 1, "__definition_n_KLzsamp": "The number of samples to use to perfrom MonteCarlo integration over z to compute the KL", "batch_norm": false, "__definition__batch_norm": "if true, do batch normalization in all layers of neural network", "twod_conv": false, "__definition__twod_conv": "if true treat the detectors as an extra image dimension otherwise as channels", "parallel_conv": false, "__definition__parallel_conv": "if true analyse detectors separately in convolutions, otherwise as channels", "report_interval": 1000, "__definition__report_interval": "interval at which to save objective function values and optionally print info during training", "n_modes": 4, "__definition__n_modes": "number of modes in Gaussian mixture model for the r1 distribution", "n_modes_q": 1, "__definition__n_modes_q": "number of modes in Gaussian mixture model for the q distribution", "n_filters_commonconv": [], "__definition__n_filters_commonconv": "number of convolutional filters to use in common convolutional network", "filter_size_commonconv": [], "__definition__filter_size_r1": "size of convolutional fitlers in common convolutional network", "n_filters_r1": [32,32,32,32], "__definition__n_filters_r1": "number of convolutional filters to use in r1 network (must be divisible by 3)", "n_filters_r2": [32,32,32,32], "__definition__n_filters_r2": "number of convolutional filters to use in r2 network (must be divisible by 3)", "n_filters_q": [32,32,32,32], "__definition__n_filters_q": "number of convolutional filters to use in q network (must be divisible by 3)", "filter_size_r1": [5,5,5,5], "__definition__filter_size_r1": "size of convolutional fitlers in r1 network", "filter_size_r2": [5,5,5,5], "__definition__filter_size_r2": "size of convolutional filters in r2 network", "filter_size_q": [5,5,5,5], "__definition__filter_size_q": "size of convolutional filters in q network", "drate": 0.0, "__definition__drate": "dropout rate to use in fully-connected layers", "maxpool_commonconv": [], "__definition__maxpool_commonconv": "size of maxpooling to use in r1 network", "conv_strides_commonconv": [], "__definition__conv_strides_commonconv": "size of convolutional stride to use in r1 network", "conv_dilations_commonconv": [], "__definition__conv_dilations_commonconv": "size of convolutional dilation to use in r1 network", "pool_strides_commonconv": [], "__definition_pool_strides_commonconv": "size of max pool stride to use in r1 network", "maxpool_r1": [1,1,1,1], "__definition__maxpool_r1": "size of maxpooling to use in r1 network", "conv_strides_r1": [1,1,1,1], "__definition__conv_strides_r1": "size of convolutional stride to use in r1 network", "conv_dilations_r1": [1,1,1,1], "__definition__conv_dilations_r1": "size of convolutional dilation to use in r1 network", "pool_strides_r1": [1,1,1,1], "__definition_pool_strides_r1": "size of max pool stride to use in r1 network", "maxpool_r2": [1,1,1,1], "__definition_maxpool_r2": "size of max pooling to use in r2 network", "conv_strides_r2": [1,1,1,1], "__definition__conv_strides_r2": "size of convolutional stride in r2 network", "conv_dilations_r2": [1,1,1,1], "__definition__conv_dilations_r2": "size of convolutional dilation to use in r2 network", "pool_strides_r2": [1,1,1,1], "__definition__pool_strides_r2": "size of max pool stride in r2 network", "maxpool_q": [1,1,1,1], "__definition__maxpool_q": "size of max pooling to use in q network", "conv_strides_q": [1,1,1,1], "__definition__conv_strides_q": "size of convolutional stride to use in q network", "conv_dilations_q": [1,1,1,1], "__definition__conv_dilations_q": "size of convolutional dilation to use in r1 network", "pool_strides_q": [1,1,1,1], "__definition__pool_strides_q": "size of max pool stride to use in q network", "ramp_start": 10000.0, "__definition__ramp_start": "starting iteration of KL divergence ramp", "ramp_end": 100000.0, "__definition__ramp_end": "ending iteration of KL divergence ramp", "save_interval": 25000, "__definition__save_interval": "number of iterations to save model", "plot_interval": 25000, "__definition__plot_interval": "number of iterations to plot validation results corner plots", "n_weights_r1": [4096,2048,1024], "__definition__n_weights_r1": "number fully-connected neurons in layers of encoders and decoders in the r1 model", "n_weights_r2": [4096,2048,1024], "__definition__n_weights_r2": "number fully-connected neurons in layers of encoders and decoders in the r2 model", "n_weights_q": [4096,2048,1024], "__definition__n_weights_q": "number fully-connected neurons in layers of encoders and decoders in the q model", "corner_labels": { "mass_1": "$m_{1}\\,(\\mathrm{M}_{\\odot})$", "mass_2": "$m_{2}\\,(\\mathrm{M}_{\\odot})$", "luminosity_distance": "$d_{\\mathrm{L}}\\,(\\mathrm{Mpc})$", "geocent_time": "$t_{0}\\,(\\mathrm{seconds})$", "phase": "$\\phi\\,(\\mathrm{rad})$", "theta_jn": "$\\Theta_{jn}\\,(\\mathrm{rad})$", "psi": "$\\psi\\,(\\mathrm{rad})$", "a_1": "${a_1}$", "a_2": "${a_2}$", "tilt_1": "$\\Theta_{1}\\,(\\mathrm{rad})$", "tilt_2": "$\\Theta_{2}\\,(\\mathrm{rad})$", "phi_12": "$\\phi_{12}\\,(\\mathrm{rad})$", "phi_jl": "$\\phi_{jl}\\,(\\mathrm{rad})$", "ra": "${\\alpha}\\,(\\mathrm{rad})$", "dec": "${\\delta}\\,(\\mathrm{rad})$" }, "__definition__corner_labels": "latex source parameter labels for plotting", "ref_geocent_time": 1126259642.5, "__definition__ref_geocent_time": "reference gps time (not advised to change this)", "training_data_seed": 1, "__definition__training_data_seed": "tensorflow training random seed number", "validation_data_seed": 2, "__definition__validation_data_seed": "tensorflow validation random seed number", "testing_data_seed": 3, "__definition__testing_data_seed": " tensorflow testing random seed number", "rand_pars": ["mass_1","mass_2","luminosity_distance","geocent_time","phase","theta_jn","psi"], "__definition__rand_pars": "parameters to randomize (those not listed here are fixed otherwise)", "inf_pars": ["mass_1","mass_2","luminosity_distance","geocent_time"], "__definition__inf_pars": "parameters to infer", "bilby_pars": ["mass_1","mass_2","luminosity_distance","geocent_time","theta_jn","psi","ra","dec","a_1","a_2","tilt_1","tilt_2","phi_12","phi_jl"], "__definition__bilby_pars": "parameters for bilby to infer", "train_set_dir": "/home/chrism/Heni_codes/train_data/", "__definition__train_set_dir": "location of training set directory", "val_set_dir": "/home/chrism/Heni_codes/val_data/", "__definition__val_set_dir": "location of validation set directory", "test_set_dir": "/home/chrism/Heni_codes/test_data/", "__definition__test_set_dir": "location of test set directory waveforms", "pp_test_set_dir":"/home/chrism/Heni_codes/pp_data/", "__definition__pp_test_set_dir":"location of the test set used for p-p plot", "pe_dir": "/home/chrism/vitamin_b/test_sets_3det_15par_256Hz/all_1_samplers/test", "__definition__pe_dir": "location of test set directory Bayesian PE samples. Set to None if you do not want to use posterior samples when training.", "samplers": ["vitamin","dynesty"], "__definition__samplers": "Samplers to use when comparing ML results (vitamin is ML approach) dynesty,ptemcee,cpnest,emcee", "figure_sampler_names": ["VItamin","Dynesty"], "__definition__figure_sampler_names": "matplotlib figure sampler labels (e.g. [ptemcee,CPNest,emcee])", "y_normscale": 36.0, "__definition__y_normscale": "arbitrary normalization factor on all time series waveforms (helps convergence in training)", "gauss_pars": ["luminosity_distance","geocent_time","theta_jn","a_1","a_2","tilt_1","tilt_2","ra","dec"], "__definition__gauss_pars": "parameters that require a truncated gaussian distribution", "vonmise_pars": ["phase","phi_12","phi_jl","psi"], "__definition__vonmises_pars": "parameters that get wrapped on the 1D parameter", "sky_pars": [], "__definition__sky_pars": "sky parameters", "psd_files": [], "__definition__psd_files": "User may specficy their own psd files for each detector. Must be bilby compatible .txt files. If specifying your own files, do so for each detector. If list is empty, default Bilby PSD values will be used for each detector.", "use_real_det_noise": false, "__definition__use_real_det_noise": "If True, use real detector noise around reference time", "use_real_events": [], "__definition__use_real_events": "list containing real ligo events to be analyzed", "make_corner_plots": true, "__definition__make_corner_plots": "if True, make corner plots", "make_kl_plot": false, "__definition__make_kl_plot": "If True, go through kl plotting function", "make_pp_plot": false, "__definition__make_pp_plot": "If True, go through pp plotting function", "make_loss_plot": true, "__definition__make_loss_plot": "If True, generate loss plot from previous plot data", "Make_sky_plot": false, "__definition__make_sky_plot": "If True, generate sky plots on corner plots", "resume_training": false, "__definition___resume_training": "if True, resume training of a model from saved checkpoint", "load_by_chunks": true, "__definition__load_by_chunks": "if True, load training samples by a predefined chunk size rather than all at once" }