diff --git a/nifty5/minimization/iteration_controllers.py b/nifty5/minimization/iteration_controllers.py index c0c2c9ca111f87e377a546a01201cc45fbe15e65..251a3ffa6ba2d1667e7fe07461fc06ce74faeb15 100644 --- a/nifty5/minimization/iteration_controllers.py +++ b/nifty5/minimization/iteration_controllers.py @@ -100,6 +100,7 @@ class GradientNormController(IterationController): self._name = name def start(self, energy): + self.energyhistory = [] self._itcount = -1 self._ccount = 0 if self._tol_rel_gradnorm is not None: @@ -128,6 +129,7 @@ class GradientNormController(IterationController): "{}: Iteration #{} energy={:.6E} gradnorm={:.2E} clvl={}" .format(self._name, self._itcount, energy.value, energy.gradient_norm, self._ccount)) + self.energyhistory.append(energy.value) # Are we done? if self._iteration_limit is not None: