Skip to content

Commit d39fa93

Browse files
authored
Update make_simple_step_size_update_policy
tfp.mcmc.make_simple_step_size_update_policy() requires num_adaptation_steps as an argument. I am not sure if the correct value is None but it gives the same results as the PyMC3 version of this notebook.
1 parent d8043bf commit d39fa93

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: Chapter1_Introduction/Ch1_Introduction_TFP.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -1273,7 +1273,7 @@
12731273
" target_log_prob_fn=unnormalized_log_posterior,\n",
12741274
" num_leapfrog_steps=2,\n",
12751275
" step_size=step_size,\n",
1276-
" step_size_update_fn=tfp.mcmc.make_simple_step_size_update_policy(),\n",
1276+
" step_size_update_fn=tfp.mcmc.make_simple_step_size_update_policy(num_adaptation_steps=None),\n",
12771277
" state_gradients_are_stopped=True),\n",
12781278
" bijector=unconstraining_bijectors))\n",
12791279
"\n",

0 commit comments

Comments
 (0)