Model can now be saved into file. Stupid mistake forgetting the quote
Limit the number of training samples due to memory constraint