From a9f3e7a19164d0808a4ad15bae54ab918b7f6cd7 Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Mon, 22 Jul 2019 21:20:40 +0200
Subject: [PATCH] Temporary: Add energyhistory to iteration controllers
---
nifty5/minimization/iteration_controllers.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/nifty5/minimization/iteration_controllers.py b/nifty5/minimization/iteration_controllers.py
index c0c2c9ca1..251a3ffa6 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:
--
GitLab