DDU for segmentation

Hi everyone!

Has anyone tried to implement the DDU approach for uncertainty quantification for segmentation?

I have checked multiple repos, but still not working for me, specifically fitting the GMM

If someone knows how to get over this, please tell me.

Thanks!