Added parameter saving to CE.  Still crashing in training.