Early stopping to avoid overfitting. Now we just need to optimize the hell out of this.