Working really nicely now, less than 200 epochs every time

Would like a better selection method though

Maybe a GA for selecting the params for my GA LOL