diff --git a/variational_inference_visualized.py b/variational_inference_visualized.py index 3531ee42b8451f0c422b996c0259e02606aa58f5..6bb0a899c9c6c1a789382ea44311f7ef854b6b76 100644 --- a/variational_inference_visualized.py +++ b/variational_inference_visualized.py @@ -36,7 +36,7 @@ def main(): dom = ift.UnstructuredDomain(1) n_samples = 20 - show_geo = True + show_geo = False scale = 10. def transformation(x,y): @@ -185,7 +185,7 @@ def main(): axs[1].set_xlabel('x') fig.tight_layout() plt.draw() - plt.pause(2. if not start else 5.) + plt.pause(2. if not start else 10.) minimizer = ift.NewtonCG( @@ -194,7 +194,7 @@ def main(): posgeo = ift.full(ham.domain, -5.) for ii in range(30): - if ii % 3 == 0: + if ii % 4 == 0: # Resample GeoVI and MGVI mgkl = ift.SampledKLEnergy(posmg, ham, n_samples, None, True) mini_samp = ift.NewtonCG(