Was going to do 15 epochs, but it crashed after 10. Seems to work though.