diff --git a/SimPEG/Directives.py b/SimPEG/Directives.py index 9e4626ca..e91ffb71 100644 --- a/SimPEG/Directives.py +++ b/SimPEG/Directives.py @@ -210,31 +210,34 @@ class SaveOutputDictEveryIteration(_SaveEveryIteration): """SaveOutputDictEveryIteration""" def initialize(self): - print "SimPEG.SaveOutputDictEveryIteration will save your inversion progress as dictionary: '###-%s.npz'" - + print "SimPEG.SaveOutputDictEveryIteration will save your inversion progress as dictionary: '###-%s.npz'"%self.fileName def endIter(self): # Save the data. - ms = self.reg.Ws * ( self.reg.mapping * (self.invProb.curModel - self.r + ms = self.reg.Ws * ( self.reg.mapping * (self.invProb.curModel - self.reg.mref) ) phi_ms = 0.5*ms.dot(ms) if self.reg.smoothModel == True: mref = self.reg.mref else: mref = 0 - mx = self.reg.Wx * ( self.reg.mapping * (self.invProb.curModel - mref) + mx = self.reg.Wx * ( self.reg.mapping * (self.invProb.curModel - mref) ) phi_mx = 0.5 * mx.dot(mx) if self.prob.mesh.dim==2: - my = self.reg.Wy * ( self.reg.mapping * (self.invProb.curModel - mr + my = self.reg.Wy * ( self.reg.mapping * (self.invProb.curModel - mref) ) phi_my = 0.5 * my.dot(my) else: phi_my = 'NaN' if self.prob.mesh.dim==3: - mz = self.reg.Wz * ( self.reg.mapping * (self.invProb.curModel - mr + mz = self.reg.Wz * ( self.reg.mapping * (self.invProb.curModel - mref) ) phi_mz = 0.5 * mz.dot(mz) else: phi_mz = 'NaN' + # Save the file as a npz + np.savez('{:03d}-{:s}'.format(self.opt.iter,self.fileName), iter=self.opt.iter, beta=self.invProb.beta, phi_d=self.invProb.phi_d, phi_m=self.invProb.phi_m, phi_ms=phi_ms, phi_mx=phi_mx, phi_my=phi_my, phi_mz=phi_mz,f=self.opt.f, m=self.invProb.curModel) + + # class UpdateReferenceModel(Parameter):