Prediction

[szuhow]

After a few adjustments to work on my Mac with M1 and with the dataset in grayscale mode, I managed to achieve the following results after 20 epochs: train loss 0.53152, validation Dice 0.60535, and segmentation results on the images (vein segmentation). I think it would be useful to have an option to set the proportion of BCE to Dice.