ayadav01:

Hi Patrice, For a segmentation model, i would suggest you to use a soft dice loss rather than using weighted nn.BCEWithLogitsLoss()


Hello Anil, thank you very much  for your suggestion. I will give it a try after fixing my computer. I am facing some hardware problems presently.