mateuszbuda / brain-segmentation-pytorch

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI
https://mateuszbuda.github.io/2017/12/01/brainseg.html
MIT License
722 stars 187 forks source link

how can I apply other loss functions #30

Closed nepersica closed 3 years ago

nepersica commented 3 years ago

I'm thankful sharing pretrained weights.. I have a question for using.

I want to use other losses(ex.BCE, focal) on training.(not dice loss.)

When I use other functions, loss didn't decrease from particular value, but dice loss didn't.

Do the pretrained weights fit on dice loss only?