Using "segmentation_models_pytorch" with Unet++

45 views Asked by At

I was trying to use segmentation_models_pytorch" with Unet++. I tried to look at the example at HuBMAP - Pytorch smp Unet++. My understanding Unet++ returns 4 outputs (as part of deep supervision). But in the example, I notice it is only returning one output. I think I'm missing something here.

0

There are 0 answers