Hey, Dusan !
I found a solution in this thread for fixing the mask problems. However, this instruction
learn = unet_learner(data, models.resnet34, loss_func = nn.BCELoss, metrics = [salt_acc], wd = 1e-2).to_fp16()
returned an error saying that the bool value of a tensor was ambigous.
Finally, I launched the model without defining a particular loss function adding simple a custom metric and it seemed to work fine !
Would you happen to have an idea about this error ?
bool value of Tensor with more than one value is ambiguous
Thanks anyway for your help !