[WIP] Inference debugging work

notebooks/bayes3d_paper/tester2.ipynb

@@ -184,16 +184,16 @@
     "        jnp.array([1-p_resample_color, p_resample_color])\n",
     "    ),\n",
     "    \"visibility_prob_kernel\": transition_kernels.DiscreteFlipKernel(\n",
-    "        resample_probability=0.05, support=jnp.array([0.0, 0.998])\n",
+    "        p_change_to_different_value=0.05, support=jnp.array([0.0, 0.998])\n",
     "    ),\n",
     "    \"depth_nonreturn_prob_kernel\": transition_kernels.DiscreteFlipKernel(\n",
-    "        resample_probability=0.2, support=jnp.array([0.002, 0.998])\n",
+    "        p_change_to_different_value=0.2, support=jnp.array([0.002, 0.998])\n",
     "    ),\n",
     "    \"depth_scale_kernel\": transition_kernels.DiscreteFlipKernel(\n",
-    "        resample_probability=0.05, support=jnp.array([0.0025, 0.01, 0.02, .1])#, .1, .4, 1.])\n",
+    "        p_change_to_different_value=0.05, support=jnp.array([0.0025, 0.01, 0.02, .1])#, .1, .4, 1.])\n",
     "    ),\n",
     "    \"color_scale_kernel\": transition_kernels.DiscreteFlipKernel(\n",
-    "        resample_probability=0.05, support=jnp.array([0.01, 0.05, 0.1, .3])#, 0.15, .3, .8])\n",
+    "        p_change_to_different_value=0.05, support=jnp.array([0.01, 0.05, 0.1, .3])#, 0.15, .3, .8])\n",
     "    ),\n",
     "\n",
     "    \"image_kernel\": img_model,\n",