Commit 521de0a ThibeauWouters
committed
1 parent e15e705 commit 521de0a Copy full SHA for 521de0a
File tree 1 file changed +22
-0
lines changed
1 file changed +22
-0
lines changed Original file line number Diff line number Diff line change
1
+ jim_default_hyperparameters = {
2
+ "seed" : 0 ,
3
+ "n_chains" : 20 ,
4
+ "num_layers" : 10 ,
5
+ "hidden_size" : [128 ,128 ],
6
+ "num_bins" : 8 ,
7
+ "local_sampler_arg" : {},
8
+ "n_walkers_maximize_likelihood" : 100 ,
9
+ "n_loops_maximize_likelihood" : 200 ,
10
+ }
11
+
12
+ jim_explanation_hyperparameters = {
13
+ "seed" : "(int) Value of the random seed used" ,
14
+ "n_chains" : "(int) Number of chains to be used" ,
15
+ "num_layers" : "(int) Number of hidden layers of the NF" ,
16
+ "hidden_size" : "List[int, int] Sizes of the hidden layers of the NF" ,
17
+ "num_bins" : "(int) Number of bins used in MaskedCouplingRQSpline" ,
18
+ "local_sampler_arg" : "(dict) Additional arguments to be used in the local sampler" ,
19
+ "rng_key_set" : "(jnp.array) Key set to be used in PRNG keys" ,
20
+ "n_walkers_maximize_likelihood" : "(int) Number of walkers used in the maximization of the likelihood with the evolutionary optimizer" ,
21
+ "n_loops_maximize_likelihood" : "(int) Number of loops to run the evolutionary optimizer in the maximization of the likelihood" ,
22
+ }
You can’t perform that action at this time.
0 commit comments