Hi @ptrblck
Yes predictions are negative, (output_batch < 0.0).any() returns 1
What should I do? The input image to a U-Net model is a set of RGB images, and training mask is a 10-channel mask corresponding to the 10 different classes that I want to segment in the image.