Commit 53f6937b authored by Chris Jewell's avatar Chris Jewell
Browse files

Switch to regular TF graph mode

Regular TF graph mode appears to be faster for the current code
implementation.
parent 3ef67e70
......@@ -76,7 +76,7 @@ def run_pipeline(global_config, results_directory, cli_options):
global_config,
)
def run_mcmc(input_file, output_file, config):
mcmc(input_file, output_file, config["Mcmc"])
mcmc(input_file, output_file, config["Mcmc"], use_autograph=True, use_xla=False)
@rf.transform(
input=run_mcmc,
......
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