add prompt to save weight data to file after training
stupid machine needs to learn better