Add save method and load method for saving and 
loading training models.