diff --git a/demos/getting_started_3.py b/demos/getting_started_3.py index 54e7a4c90a83445db798b6ad5771061797c40c0a..a5e2063433e5edd8d5231c08fc99f7d94fc9948a 100644 --- a/demos/getting_started_3.py +++ b/demos/getting_started_3.py @@ -91,7 +91,7 @@ if __name__ == '__main__': # number of samples used to estimate the KL N_samples = 20 for i in range(2): - KL = ift.KL_Energy(position, H, N_samples) + KL = ift.KL_Energy(position, H, N_samples, want_metric=True) KL, convergence = minimizer(KL) position = KL.position