# lvl = logging.DEBUG
# logging.basicConfig(level=lvl, format='%(message)s', stream=sys.stdout)
# Optimize the model.
for step in range(50):
print('Tractor params:')
tractor.printThawedParams()
dlnp,X,alpha = tractor.optimize(damp=1.)
print('dlnp', dlnp)
print('galaxy:', moggal)
#print('Mog', moggal.mog.getParams())
if dlnp == 0:
break
# Plot the model as we're optimizing...
mod = tractor.getModelImage(0)
chi = (tim.getImage() - mod) * tim.getInvError()
plt.clf()
plt.subplot(1,2,1)
plt.imshow(mod, interpolation='nearest', origin='lower')
plt.title('Model')
plt.subplot(1,2,2)
mx = np.abs(chi).max()
plt.imshow(chi, interpolation='nearest', origin='lower',
vmin=-mx, vmax=mx)
plt.colorbar()
plt.title('Chi residuals')
plt.suptitle('MoG model after optimization step %i' % step)
plt.savefig('mod-o%02i.png' % step)
