it dose't work in my customed dataset kvasir-seg

[ssxfc]

i add kvasir-seg dataset(1000 samples) to source code and modify some below:

kvasir-seg consists of img and mask with 3 channels and 1 channel respectively.
and in my settings, sigma is learnable. The result is wrong above after 115,000 iters.
please tell me what should i do?
this is my repo[diff_kvasir]: https://github.com/ssxfc/Diffusion-based-Segmentation/tree/diff_kvasir
appendix:
uncentainty metric of top img in this comments