Can you train STANet? When I train it,Imeet a error.

[wangjin1234567891]

[李开宇]

@LinoSun

[LinoSun]

Could plz show more code.

[wangjin1234567891]

I trian Unet and it's ok .I just change the model as following picture shows

[LinoSun]

We still need debug STANet later.If you want use it now. You can try to use BCLLoss or other loss function for STANet because of STANet model only forward one channel distance map.

[wangjin1234567891]

Ok, thans a lot.