fix early stopping
fix early stopping