Commit e66e8b5e authored by Martin Reinecke's avatar Martin Reinecke
Browse files


parent 1f2dcab4
Pipeline #71405 passed with stages
in 17 minutes and 1 second
......@@ -257,7 +257,7 @@ class MetricGaussianKL(Energy):
res = res + tuple(-item for item in res)
return res
def _unscaled_metric_sample(self, from_inverse=False, dtype=np.float64):
def _metric_sample(self, from_inverse=False, dtype=np.float64):
if from_inverse:
raise NotImplementedError()
lin = self._lin.with_want_metric()
......@@ -269,7 +269,4 @@ class MetricGaussianKL(Energy):
if self._mirror_samples:
samp = samp + self._hamiltonian(lin-v).metric.draw_sample(from_inverse=False, dtype=dtype)
return _allreduce_sum_field(self._comm, samp)
def _metric_sample(self, from_inverse=False, dtype=np.float64):
return self._unscaled_metric_sample(from_inverse, dtype)/self._n_eff_samples
return _allreduce_sum_field(self._comm, samp)/self._n_eff_samples
Supports Markdown
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