Early Stopping

Stop when validation loss stops improving.