Commit 73fcac5c authored by Chris Jewell's avatar Chris Jewell
Browse files

Added option

parent 5630746b
......@@ -54,7 +54,7 @@ def _get_window_sizes(num_adaptation_steps):
return first_window_size, slow_window_size, last_window_size
@tf.function
@tf.function(jit_compile=False)
def _fast_adapt_window(
num_draws,
joint_log_prob_fn,
......@@ -112,7 +112,7 @@ def _fast_adapt_window(
return draws, trace, step_size, weighted_running_variance
@tf.function
@tf.function(jit_compile=False)
def _slow_adapt_window(
num_draws,
joint_log_prob_fn,
......@@ -181,7 +181,7 @@ def _slow_adapt_window(
)
@tf.function # (experimental_compile=True)
@tf.function(jit_compile=False)
def _fixed_window(
num_draws,
joint_log_prob_fn,
......
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