add the early stop parameter

zackli4ai changed pull request status to merged

Sign up or log in to comment