HiLab-git / SSL4MIS

Semi Supervised Learning for Medical Image Segmentation, a collection of literature reviews and code implementations.
MIT License
2.13k stars 381 forks source link

about URPC train #58

Closed Roll77-luo closed 2 years ago

Roll77-luo commented 2 years ago

I reproduced the URPC code, and when I'm training my model, the uncertain values(consistency_loss_aux1,....) gets smaller and smaller, and it approaches infinity, lead my model train failure

Roll77-luo commented 2 years ago

My model out is two channels,Only foreground and background does it matter