Question about the beta schedule

[Calvin-Pang]

Hi, thanks for publishing the code! I am trying to implement your code on other MRI modalities, whose noise levels may be less series than DWI images. However, I found the matched states in Stage 2 are quite small (around 0~50). Do you have any insights on how to use your method on less noisy images, or have you done these kinds of experiments? Thanks so much!