Saving training models to file