Skip to content
Snippets Groups Projects
Commit a9f3e7a1 authored by Philipp Arras's avatar Philipp Arras
Browse files

Temporary: Add energyhistory to iteration controllers

parent d8aa2644
Branches
Tags
2 merge requests!349Fix mpi,!333Operator spectra
Pipeline #52277 passed
...@@ -100,6 +100,7 @@ class GradientNormController(IterationController): ...@@ -100,6 +100,7 @@ class GradientNormController(IterationController):
self._name = name self._name = name
def start(self, energy): def start(self, energy):
self.energyhistory = []
self._itcount = -1 self._itcount = -1
self._ccount = 0 self._ccount = 0
if self._tol_rel_gradnorm is not None: if self._tol_rel_gradnorm is not None:
...@@ -128,6 +129,7 @@ class GradientNormController(IterationController): ...@@ -128,6 +129,7 @@ class GradientNormController(IterationController):
"{}: Iteration #{} energy={:.6E} gradnorm={:.2E} clvl={}" "{}: Iteration #{} energy={:.6E} gradnorm={:.2E} clvl={}"
.format(self._name, self._itcount, energy.value, .format(self._name, self._itcount, energy.value,
energy.gradient_norm, self._ccount)) energy.gradient_norm, self._ccount))
self.energyhistory.append(energy.value)
# Are we done? # Are we done?
if self._iteration_limit is not None: if self._iteration_limit is not None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment