typo?

[JoJoistheBestOne]

[SimonWXW]

hi, thanks for pointing out the typo. it should be:

for the first 40k steps of Stages 1 and 2, followed by a decay to $3\times 10^{-5}$ for the final 45k steps. For Stage 3, it is reduced to $1\times 10^{-5}$.

we will also update the paper next week