Commit 3fbe334c authored by Martin Reinecke's avatar Martin Reinecke

fixes

parent 6bbeea6f
......@@ -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
......
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