bugs related to the stop probability

[shixuansun]

Hi, thanks for your open source.

We find that despite that the command line provides the parameter to set the stop probability, your source code sets it to 0.15 without taking the input.