Commit ef4fd98f authored by Chris Jewell's avatar Chris Jewell
Browse files

Merge branch 'posterior_chunks_fix' into 'master'

Created lower bound for posterior events chunk size

See merge request !21
parents 65bd815a ea557405
......@@ -280,7 +280,10 @@ def mcmc(data_file, output_file, config, use_autograph=False, use_xla=True):
samples[1][:, 1:],
(NUM_BURST_SAMPLES, samples[1].shape[1] - 1),
),
"events": (samples[2], (NUM_BURST_SAMPLES, 32, 32, 1)),
"events": (
samples[2],
(NUM_BURST_SAMPLES, min(32, events.shape[1]), 32, 1),
),
},
results_dict=results,
num_samples=NUM_SAVED_SAMPLES,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment