Overfit, epoch 3-4 prob best