From 6b78a8617c3802d720209b1178f5ba5bc7273562 Mon Sep 17 00:00:00 2001 From: rowanc1 Date: Thu, 9 Oct 2014 15:28:59 -0700 Subject: [PATCH] save model using numpy --- SimPEG/Directives.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/SimPEG/Directives.py b/SimPEG/Directives.py index 6a492a50..e84a7d39 100644 --- a/SimPEG/Directives.py +++ b/SimPEG/Directives.py @@ -145,6 +145,24 @@ class BetaSchedule(InversionDirective): self.invProb.beta /= self.coolingFactor + +class SaveModelEveryIteration(InversionDirective): + """SaveModelEveryIteration""" + + @property + def modelName(self): + if getattr(self, '_modelName', None) is None: + from datetime import datetime + self._modelName = 'inversionModel-%s'%datetime.now().strftime('%Y-%m-%d') + return self._modelName + @modelName.setter + def modelName(self, value): + self._modelName = value + + def endIter(self): + np.save('%03d-%s' % (self.opt.iter, self.modelName), self.opt.xc) + + # class UpdateReferenceModel(Parameter): # mref0 = None