reproduction of this https://arxiv.org/pdf/2407.03297

cut-head is obtained by scheduler(step=1002)[1:-1] to avoid some extreme high/low value which "may" cause problem. If you meet any weird result with normal one, try cut-head.

red line means sigma=1