add save weights argument

[adityashrm21]

Add an argument to argument parser for saving the weights after the training process. The saved weights could be directly used for recommendations.